search

VDIGPKU / CBNet_caffe

Composite Backbone Network (AAAI20)
Apache License 2.0
408 stars 78 forks source link

Any tips for finetuning on private datasets? #16

Open alen-mask opened 4 years ago

alen-mask commented 4 years ago

Hi @PKUbahuangliuhe ,

Great work! So fay as i know, cbnet is the sota on coco object detection tasks. Do you have any plans to support finetuning on other datasets(different number of object classes) from your pretrained model? Any tips or reference code here?

Many thanks!

PKUbahuangliuhe commented 4 years ago

Hi @PKUbahuangliuhe ,

Great work! So fay as i know, cbnet is the sota on coco object detection tasks. Do you have any plans to support finetuning on other datasets(different number of object classes) from your pretrained model? Any tips or reference code here?

Many thanks!

you can convert the format of your data to COCO.Note that the number class should also be modified,and just train it.

alen-mask commented 4 years ago

Hi @PKUbahuangliuhe , Great work! So fay as i know, cbnet is the sota on coco object detection tasks. Do you have any plans to support finetuning on other datasets(different number of object classes) from your pretrained model? Any tips or reference code here? Many thanks!

you can convert the format of your data to COCO.Note that the number class should also be modified,and just train it.

Thanks very much, guess i should read the training code again.