Есть ли способ запретить 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
    • Model2 /
  • runTuriCreate.py

На данный момент Turicreate собирается в TrainingData, что я и ожидал... но он использует все подкаталоги, а также каталоги, которые я хочу моделировать.

Есть ли способ сделать это?

РЕДАКТИРОВАТЬ: я обнаружил, что вы можете добавить recursive=False к вызову функции load_images. Проблема в том, что я использую рекурсию... только рекурсию строго с одним уровнем глубины.

0 ответов

Другие вопросы по тегам