xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.19k stars 189 forks source link

Transfer Learning #68

Closed shoxa-mir closed 2 years ago

shoxa-mir commented 2 years ago

I want to apply CenterNet2 COCO-dataset pretrained weights to train my own custom data. My custom data has 9 classes and 7 of them (person, bicycle, car, motorcycle, bus, traffic light, truck) are similar with the COCO dataset. Can you give any instructions for transfer learning using CenterNet2?

Thank you!