有没有办法防止python turicreate使用子目录作为训练数据?

时间:2018-08-01 21:21:35

标签: python turi-create

我正在使用...获取训练数据

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 个答案:

没有答案