PaddleDetection使用问题01_COCODataSet_object_has_no_attribute

最新推荐文章于 2024-05-18 01:24:11 发布

张欣-男

最新推荐文章于 2024-05-18 01:24:11 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： PaddlePaddle 文章标签： PaddleDetection 深度学习目标检测

本文链接：https://blog.youkuaiyun.com/sdlypyzq/article/details/124164805

PaddlePaddle 专栏收录该内容

8 篇文章

订阅专栏

本文档讲述了在使用PaddleDetection进行模型预测时遇到的错误，原因在于TestDataset配置中误将ImageFolder设置为COCODataSet。解决方案是修正配置，确保EvalDataset使用COCODataSet，同时提供正确的训练和验证集配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 问题

Traceback (most recent call last):
  File "tools/infer.py", line 177, in <module>
    main()
  File "tools/infer.py", line 173, in main
    run(FLAGS, cfg)
  File "tools/infer.py", line 134, in run
    save_txt=FLAGS.save_txt)
  File "/home/zhangxin/github/PaddleDetection/ppdet/engine/trainer.py", line 541, in predict
    self.dataset.set_images(images)
AttributeError: 'COCODataSet' object has no attribute 'set_images'

2 解决办法：

错误原因是TestDataset配置错了，在改配置文件时把ImageFolder错改成了COCODataSet。

metric: COCO
num_classes: 80

TrainDataset:
  !COCODataSet
    image_dir: train2017
    anno_path: annotations/instances_train2017.json
    dataset_dir: dataset/coco
    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']

EvalDataset:
  !COCODataSet
    image_dir: val2017
    anno_path: annotations/instances_val2017.json
    dataset_dir: dataset/coco

TestDataset:
  !ImageFolder
    anno_path: annotations/instances_val2017.json
    dataset_dir: dataset/coco