用RT-DETR训练AI-TOD

[todesti2]

作者您好！我在使用RT-DETR-R50训练AI-TOD，出现报错： [12/14 20:28:51] ppdet.data.source.coco WARNING: Illegal image file: /root/autodl-tmp/AI-TOD/train/3b444d792.png, and it will be ignored Traceback (most recent call last): File "tools/train.py", line 183, in <module> main() File "tools/train.py", line 179, in main run(FLAGS, cfg) File "tools/train.py", line 126, in run trainer = Trainer(cfg, mode='train') File "/root/RT-DETR/rtdetr_paddle/ppdet/engine/trainer.py", line 77, in __init__ self.loader = create('{}Reader'.format(capital_mode))( File "/root/RT-DETR/rtdetr_paddle/ppdet/data/reader.py", line 168, in __call__ self.dataset.parse_dataset() File "/root/RT-DETR/rtdetr_paddle/ppdet/data/source/coco.py", line 241, in parse_dataset assert ct > 0, 'not found any coco record in %s' % (anno_path) AssertionError: not found any coco record in /root/autodl-tmp/AI-TOD/annotations/aitod_trainval_v1.json

我的路径没有问题，数据集也是可用的，这是为什么呢？

是因为ppdet/data/source/category.py中没有AI-TOD数据集的类说明嘛？

[lyuwenyu]

是coco的format嘛 另外图片都是png结尾的嘛

[todesti2]

是coco的format嘛 另外图片都是png结尾的嘛

是的！是COCO的格式，图片也都是png结尾，你看到的上面的图片信息报错只是其中一条，上面还有很多这样的信息。 不知道为什么，在其他任务上都可以跑这个AI-TOD

[lyuwenyu]

1. 先判断下 /root/autodl-tmp/AI-TOD/train/3b444d792.png 这路径对嘛
2. 要不用pytorch的跑一下试一试 用法一样的改一致数据集的路径就行了 看看是数据本身的问题还是代码的问题

[todesti2]

1. 先判断下 /root/autodl-tmp/AI-TOD/train/3b444d792.png 这路径对嘛
2. 要不用pytorch的跑一下试一试 用法一样的改一致数据集的路径就行了 看看是数据本身的问题还是代码的问题

不对，train文件下面还有一个images，images下面才是图片。我不知道为什么它加载出来没有images（平常在配置中也是不用写images的）

[lyuwenyu]

在这个位置改下 train -> train/images 就行了

[todesti2]

在这个位置改下 train -> train/images 就行了

多谢，改了这里确实成功了欸！