search

longzw1997 / Open-GroundingDino

This is the third party implementation of the paper Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection.
MIT License
419 stars 65 forks source link

VG数据集格式 #70

Closed HMY391 closed 5 months ago

HMY391 commented 5 months ago

请问项目中提到的odvg格式，里面detection和grounding是在同一个文件下的吗？还是说可以准备多个数据集，一个放detection的OD格式，一个放grounding的VG格式，然后在datasets_mixed_odvg.json写上就行？因为目前只想做目标检测任务，所以第一次只准备了COCO转换成odvg格式的数据（就是里面没有grounding这个键），然后训练时候报错KeyError: 'grounding'找不到这个键。那这意思是数据集里面必须还要有grounding那部分吗？

HMY391 commented 5 months ago

问题解决了！之前误以为一个类就不需要label_map了，后来看代码发现如果没有这个，他会认为你的数据是VG格式，所以才会出现KeyError: 'grounding'这个问题