TF_CenterNet

Introduction

This my implementation of CenterNet(Objects as Points) in pure TensorFlow.You can refer to the official code.Because I use light backbone without DCN, this implementation doesn't work as well as official version. The main special features of this repo inlcude:

tf.data pipeline
light backbone：resnet18, mobilenetV2
all codes were writen by pure TensorFlow ops (no keras or slim)
support training on your own dataset.

Requirements

python3
tensorflow>=1.12
opencv-python
tqdm

Train on voc dataset

1. Make dataset file

Download Pascal VOC Dataset and reorganize the directory as follows:

VOC
├── test
|    └──VOCdevkit
|        └──VOC2007
└── train
     └──VOCdevkit
         └──VOC2007
         └──VOC2012

Generate ./data/dataset/voc_train.txt and ./data/dataset/voc_test.txt, some codes of this part are from yolov3:

$ cd ./data/dataset
$ python voc_annotation.py --data_path D:/dataset/VOC

2. Download pre-train weights

You can get pre-train weights of resnet or mobilenet from my other repo. Put npy file in pretrained_weights folder.

3. Modify `cfg.py` and run `train.py`

$ python train.py

4. Inference

Update ckpt_path in inference.py,and run demo:

$ python inference.py

The result for the example images should look like:

5.Visualization

$ tensorboard --logdir=./log

Reference

[1] official code and paper
[2] YunYang1994's YOLOv3 repo

QuanixnYang/TF_CenterNet