Train on custom dataset

[Arashzarindast]

Dear author, I am trying to train your model on my custom dataset i am using following command !python src/train.py with deformable tracking resume=r50_deformable_detr-checkpoint.pth output_dir=models/custom_dataset_train_deformable mot_path=data/custom_dataset train_split=train val_split=val epochs=20 i get following error message

frame_id = self.coco.imgs[idx]['frame_id'] KeyError: 'frame_id'

[JackWoo0831]

I think you should convert your own dataset to COCO-style first, so you can get the key of 'frame_id'

[Abrahamon]

Hi @atousaz, Check your dataset. It might have the key 'image_id', instead of 'frame_id'. Some tools can do that. Such as CocoAnnotation.

[timmeinhardt]

There is a difference between image_id and frame_id. Please check the generate_coco_from_mot.py to understand each field. The latter does not exist in the original COCO annotations and is related to the position of the image as a frame in a video sequence.