tm924222

> dataset class，然后指定相应的类别。 如何新建类别呢

> 对于训练中的验证报出该错误： > > ```python > data['category_id'] = self.cat_ids[label] > IndexError: list index out of range > ``` > > 作者的建议是仿照创建一个自己的Dataset类。具体地就得自己研究尝试了，思路是先建立和继承前者的（名字自己随便取的），再在各自的中引入。`CocoDataset``BomoDataset``YoloWorldBomoDataset``__init__.py` > > 具体步骤如下： > > 步骤1.在中定义自己的类。`anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmdet/datasets/coco.py``BomoDataset` > > 1....

> [**init**](https://github.com/AILab-CVC/YOLO-World/blob/4d90f458c1d0de310643b0ac2498f188c98c819c/yolo_world/datasets/__init__.py#L12)里写进去 mmdet下面的datasets的init和mmyolo下面的datasets的init我都写了。哥们你的成功了吗

> 或者可考虑 > > ``` > dataset=dict( > type='YOLOv5CocoDataset', > metainfo=dict(classes=your_classes), > ..... > ``` 这个我也用了，后面两个的损失都是0

> mask-refine 感谢您的指导，我去修改一下

> 非常感谢你们团队提供的开源数据集，我也在上个月收到了数据集，但我发现test部分的Annotation文件是空的，不知是不是打包数据集时忘了加进去 您好，我申请了好久作者都没回复我，请问这边方便发我一份吗？[email protected]

> 嗨，你能给我一个示例注释文件吗？ ok ![Uploading 6d6eb4e01c7d6b35f8e3572704ef98f.png…]()

> Hi @tm924222, could you provide me a sample annotation file? "date_created":"2024-05-21T03:03:48+00:00"},"licenses":[{"id":1,"url":"","name":"Unknown"}],"categories":[{"id":0,"name":"1","supercategory":"none"},{"id":1,"name":"missing_hole","supercategory":"1"},{"id":2,"name":"mouse_bite","supercategory":"1"},{"id":3,"name":"open_circuit","supercategory":"1"},{"id":4,"name":"short","supercategory":"1"},{"id":5,"name":"spur","supercategory":"1"},{"id":6,"name":"spurious_copper","supercategory":"1"}],"images":[{"id":0,"license":1,"file_name":"05_open_circuit_03_jpg.rf.e13d9485402af4c1c00b7552ad6c9f01.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":1,"license":1,"file_name":"01_missing_hole_18_jpg.rf.e200e3f4e219e50d92374113738076e2.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":2,"license":1,"file_name":"11_missing_hole_10_jpg.rf.e6075768b759eef74f0d70219705da33.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":3,"license":1,"file_name":"08_missing_hole_09_jpg.rf.e1934e5d1f7f6018df3389b0794b9bf5.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":4,"license":1,"file_name":"11_missing_hole_09_jpg.rf.e0a8581b82eec73303a5d191aed8b129.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":5,"license":1,"file_name":"06_missing_hole_01_jpg.rf.e31c89a5744cc949516d0b221894836b.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":6,"license":1,"file_name":"05_missing_hole_02_jpg.rf.e04b6d11f195420bd2edc14b64ba7f99.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},{"id":7,"license":1,"file_name":"04_short_06_jpg.rf.e11f93bd670a55a875be0720cd002d6a.jpg","height":640,"width":640,"date_captured":"2024-05-21T03:03:48+00:00"},

> ```json > name":"1","supercat > ``` 感谢您的指导，恍然大悟了，我再去修改一下

> Hi @tm924222, how about the loss now? 我可能没改好，还是0