PaddleDetection使用问题01_COCODataSet_object_has_no_attribute
Posted 张欣-男
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了PaddleDetection使用问题01_COCODataSet_object_has_no_attribute相关的知识，希望对你有一定的参考价值。
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
以上是关于PaddleDetection使用问题01_COCODataSet_object_has_no_attribute的主要内容，如果未能解决你的问题，请参考以下文章
如何将PaddleDetection模型在树莓派4B上部署？