Megvii-BaseDetection / YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Apache License 2.0
9.4k stars 2.2k forks source link

训练coco格式的数据集,出现以下这种情况 #421

Open njauwhr opened 3 years ago

njauwhr commented 3 years ago

2021-08-09 13:51:54 | ERROR | yolox.core.launch:91 - An error has been caught in function 'launch', process 'MainProcess' (4300), thread 'MainThread' (140681679290560): Traceback (most recent call last):

File "/home/warcraft/whrproject/YOLOX-main/tools/train.py", line 124, in args=(exp, args), │ └ Namespace(batch_size=32, ckpt='/home/warcraft/whrproject/YOLOX-main/yolox_l.pth', devices=0, dist_backend='nccl', dist_url=No... └ ╒══════════════════╤═════════════════════════════════════════════════════════════════════════════════════════════════════════...

File "/home/warcraft/whrproject/YOLOX-main/yolox/core/launch.py", line 91, in launch main_func(*args) │ └ (╒══════════════════╤════════════════════════════════════════════════════════════════════════════════════════════════════════... └ <function main at 0x7ff282609cb0>

File "/home/warcraft/whrproject/YOLOX-main/tools/train.py", line 103, in main trainer.train() │ └ <function Trainer.train at 0x7ff27c2cee60> └ <yolox.core.trainer.Trainer object at 0x7ff27aeea310>

File "/home/warcraft/whrproject/YOLOX-main/yolox/core/trainer.py", line 68, in train self.before_train() │ └ <function Trainer.before_train at 0x7ff278cc84d0> └ <yolox.core.trainer.Trainer object at 0x7ff27aeea310>

File "/home/warcraft/whrproject/YOLOX-main/yolox/core/trainer.py", line 149, in before_train no_aug=self.no_aug, │ └ False └ <yolox.core.trainer.Trainer object at 0x7ff27aeea310>

File "/home/warcraft/whrproject/YOLOX-main/yolox/exp/yolox_base.py", line 100, in get_data_loader max_labels=50,

File "/home/warcraft/whrproject/YOLOX-main/yolox/data/datasets/coco.py", line 49, in init self.annotations = self._load_coco_annotations() │ │ └ <function COCODataset._load_coco_annotations at 0x7ff278cce950> │ └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50> └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50>

File "/home/warcraft/whrproject/YOLOX-main/yolox/data/datasets/coco.py", line 58, in _load_coco_annotations return [self.load_anno_from_ids(_ids) for _ids in self.ids] │ │ │ └ ['0000002_00005_d_0000014', '0000002_00448_d_0000015', '0000003_00231_d_0000016', '0000007_04999_d_0000036', '0000007_05499_d... │ │ └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50> │ └ <function COCODataset.load_anno_from_ids at 0x7ff278ccecb0> └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50>

File "/home/warcraft/whrproject/YOLOX-main/yolox/data/datasets/coco.py", line 58, in return [self.load_anno_from_ids(_ids) for _ids in self.ids] │ │ │ │ │ └ ['0000002_00005_d_0000014', '0000002_00448_d_0000015', '0000003_00231_d_0000016', '0000007_04999_d_0000036', '0000007_05499_d... │ │ │ │ └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50> │ │ │ └ '0000002_00005_d_0000014' │ │ └ '0000002_00005_d_0000014' │ └ <function COCODataset.load_anno_from_ids at 0x7ff278ccecb0> └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50>

File "/home/warcraft/whrproject/YOLOX-main/yolox/data/datasets/coco.py", line 61, in load_anno_from_ids imann = self.coco.loadImgs(id)[0] │ │ │ └ '0000002_00005_d_0000014' │ │ └ <function COCO.loadImgs at 0x7ff278cce290> │ └ <pycocotools.coco.COCO object at 0x7ff270374790> └ <yolox.data.datasets.coco.COCODataset object at 0x7ff270374b50>

File "/home/warcraft/anaconda3/envs/whrtorch/lib/python3.7/site-packages/pycocotools/coco.py", line 265, in loadImgs return [self.imgs[id] for id in ids] │ │ └ '0000002_00005_d_0000014' │ └ {'0000002_00005_d_0000014': {'file_name': '0000002_00005_d_0000014.jpg', 'height': 540, 'width': 960, 'id': '0000002_00005d... └ <pycocotools.coco.COCO object at 0x7ff270374790> File "/home/warcraft/anaconda3/envs/whrtorch/lib/python3.7/site-packages/pycocotools/coco.py", line 265, in return [self.imgs[id] for id in ids] │ │ │ └ '0' │ │ └ '0' │ └ {'0000002_00005_d_0000014': {'file_name': '0000002_00005_d_0000014.jpg', 'height': 540, 'width': 960, 'id': '0000002_00005d... └ <pycocotools.coco.COCO object at 0x7ff270374790>

KeyError: '0'

tongalitong commented 3 years ago

same error

Abigale-Xin commented 2 years ago

same error, have U solved the problem? @njauwhr

Spang12357 commented 11 months ago

我也没解决,有会的吗