CenterNet Implementation

Training

python3 train.py

TODO

• start from pretrained resnet
• measure COCO metrics e.g. mAP
• visualize the result in notebook
• manage depedency with uv
• handle flipped bbox min and max

Reference Implementations and Resources

• CenterNet Official Implementation
• MMDetection CenterNet Head
• Alternative CenterNet Implementation
• CenterNet Paper
• CornerNet-Lite Utils
• COCO Dataset on Hugging Face
• MMDetection CenterNet Head Implementation
• CornerNet Keypoint Utils