我正在使用...获取我的训练数据
turicreate.image_analysis.load_images('TrainingData', with_path=True)
TrainingData/有我所有的训练数据，使用文件夹名称作为标签......
data['label'] = data['path'].apply(lambda path: os.path.dirname(path.split('/')[-1]))
这得到了最后一个目录......问题是 turicreate 比我预期的要深入。
目录结构如下所示...
- 训练数据/
- 型号1/
- ADirectory/ // <== 这里有问题（我想忽略这个目录）
- img1_s.jpg
- img1.jpg
- img2.jpg
- ADirectory/ // <== 这里有问题（我想忽略这个目录）
- 型号2/
- 型号1/
- 运行TuriCreate.py
目前 turicreate 正在进入 TrainingData ，这是我所期望的......但它正在使用所有子目录以及我想要建模的目录。
有没有办法做到这一点？
编辑：我发现您可以将 recursive=False 添加到 load_images 函数调用中。这个问题是我正在使用递归......只是严格地使用一个深度级别的递归。