Есть ли способ запретить Python Turicreate использовать подкаталоги в качестве обучающих данных?
Я получаю данные о тренировках, используя...
turicreate.image_analysis.load_images('TrainingData', with_path=True)
TrainingData / содержит все мои тренировочные данные, используя имена папок в качестве меток...
data['label'] = data['path'].apply(lambda path: os.path.dirname(path.split('/')[-1]))
Это последний каталог... проблема в том, что turicreate идет глубже, чем я ожидаю.
Структура каталогов выглядит следующим образом...
- TrainingData /
- Model1 /
- ADirectory / // <== вот проблема (я хочу игнорировать этот каталог)
- img1_s.jpg
- img1.jpg
- img2.jpg
- ADirectory / // <== вот проблема (я хочу игнорировать этот каталог)
- Model2 /
- Model1 /
- runTuriCreate.py
На данный момент Turicreate собирается в TrainingData, что я и ожидал... но он использует все подкаталоги, а также каталоги, которые я хочу моделировать.
Есть ли способ сделать это?
РЕДАКТИРОВАТЬ: я обнаружил, что вы можете добавить recursive=False к вызову функции load_images. Проблема в том, что я использую рекурсию... только рекурсию строго с одним уровнем глубины.