Custom dataset

[john09282922]

Question

Dear authors,

Thanks for nice works,
Is it possible to fine tune model with custom dataset? if so, only follow coco format?

Thanks

Additional

No response

[xiuqhou]

Hi @john09282922 sorry for late reply.

To fine tune custom datasets, there are two things you need to do:

• Make sure the dataset follows coco format.
• num_classes of model_config is default to 91. If your dataset has more classes, please change it to a number larger than the max_class_id of your dataset. (For example, the max_class_id of COCO is 90, so we set num_classes to 91)