Training on dataset without segmentation masks

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:

MIT License

7.28k stars 1.93k forks source link

Training on dataset without segmentation masks #764

Open WingRS opened 4 years ago

WingRS commented 4 years ago

Hi! I want to try out the CenterNet on my custom dataset, which has only bounding boxex. Is this possible? If yes, could you point me what to change ? Thanks.

xingyizhou commented 4 years ago

It is completely possible. We didn't use segmentation anywhere, no code needs to be changed.