search

RainBowLuoCS / DiffusionTrack

[AAAI 2024] DiffusionTrack: Diffusion Model For Multi-Object Tracking. DiffusionTrack is the first work to employ the diffusion model for multi-object tracking by formulating it as a generative noise-to-tracking diffusion process.
Other
168 stars 7 forks source link

yolox error #3

Closed David1-git closed 2 months ago

David1-git commented 1 year ago

ERROR | yolox.core.launch:90 - An error has been caught in function 'launch', process 'MainProcess' (44024), thread 'MainThread' (540):

RainBowLuoCS commented 1 year ago

Can you offer more details about the error.

David1-git commented 1 year ago

Can you offer more details about the error.

2023-09-15 19:49:49 | ERROR | yolox.core.launch:90 - An error has been caught in function 'launch', process 'MainProcess' (14291), thread 'MainThread' (140086303749952): Traceback (most recent call last):

File "/home/valca509/桌面/LRZ/DiffusionTrack/tools/train.py", line 129, in launch( └ <function launch at 0x7f684654df70>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/core/launch.py", line 90, in launch main_func(*args) │ └ (╒══════════════════╤════════════════════════════════════════════════════════════════════════════════════════════════════════... └ <function main at 0x7f6821f14700>

File "/home/valca509/桌面/LRZ/DiffusionTrack/tools/train.py", line 110, in main trainer.train() │ └ <function Trainer.train at 0x7f6821f4a940> └ <yolox.core.trainer.Trainer object at 0x7f6821f1cac0>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/core/trainer.py", line 71, in train self.before_train() │ └ <function Trainer.before_train at 0x7f6821f6e1f0> └ <yolox.core.trainer.Trainer object at 0x7f6821f1cac0>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/core/trainer.py", line 148, in before_train self.train_loader = self.exp.get_data_loader( │ │ │ └ <function Exp.get_data_loader at 0x7f6821f174c0> │ │ └ ╒══════════════════╤═════════════════════════════════════════════════════════════════════════════════════════════════════════... │ └ <yolox.core.trainer.Trainer object at 0x7f6821f1cac0> └ <yolox.core.trainer.Trainer object at 0x7f6821f1cac0>

File "/home/valca509/桌面/LRZ/DiffusionTrack/exps/example/mot/yolox_x_diffusion_det_mot20.py", line 48, in get_data_loader dataset = MOTDataset( └ <class 'yolox.data.datasets.mot.MOTDataset'>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/data/datasets/mot.py", line 45, in init self.annotations = self._load_coco_annotations() │ │ └ <function MOTDataset._load_coco_annotations at 0x7f6821f67d30> │ └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700> └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/data/datasets/mot.py", line 56, in _load_coco_annotations return [self.load_anno_from_ids(index,_ids) for index,_ids in enumerate(self.ids)] │ │ │ └ [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 3... │ │ └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700> │ └ <function MOTDataset.load_anno_from_ids at 0x7f6821f67dc0> └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/data/datasets/mot.py", line 56, in return [self.load_anno_from_ids(index,_ids) for index,_ids in enumerate(self.ids)] │ │ │ │ │ │ │ └ [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 3... │ │ │ │ │ │ └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700> │ │ │ │ │ └ 1 │ │ │ │ └ 0 │ │ │ └ 1 │ │ └ 0 │ └ <function MOTDataset.load_anno_from_ids at 0x7f6821f67dc0> └ <yolox.data.datasets.mot.MOTDataset object at 0x7f6821efc700>

File "/home/valca509/桌面/LRZ/DiffusionTrack/yolox/data/datasets/mot.py", line 60, in load_anno_from_ids width = im_ann["width"] └ {'file_name': 'mot20_train/MOT20-01/img1/000001.jpg', 'id': 1, 'frame_id': 1, 'prev_image_id': -1, 'next_image_id': 2, 'video...

KeyError: 'width'

RainBowLuoCS commented 1 year ago

Make sure run the tools/convert_mot20_to_coco.py file to preprocess the mot20 dataset before model training. We can see the width and height property writing from line 57 to 63 in above format converting file.

David1-git commented 1 year ago

Make sure run the tools/convert_mot20_to_coco.py file to preprocess the mot20 dataset before model training. We can see the width and height property writing from line 57 to 63 in above format converting file.

I have run the file you mentioned, and the same error is still reported