有没有办法阻止pythonturicreate使用子目录作为训练数据?

2024-06-28 19:54:59 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用

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/
      • 目录///<;==问题是(我想忽略这个目录)
        • img1\美国.jpg
      • 图片1.jpg
      • 图片2.jpg
    • 型号2/
  • runTuriCreate.py文件

目前,turicreate正在进入培训数据,这是我所期望的。。。但是它使用了所有的子目录以及我想要建模的目录

有办法吗

EDIT:我发现可以将recursive=False添加到load\u images函数调用中。问题是我在使用递归。。。严格来说,只需一个深度级别的递归即可。


Tags: 数据pathimage目录truedatawith图片