FCOS (Fully Convolutional One-Stage Object Detection) implementation in Keras and Tensorflow
This is an implementation of FCOS on keras and Tensorflow. The project is based on fizyr/keras-retinanet and tianzhi0549/FCOS. Thanks for their hard work.
backbone | mAP50 |
---|---|
resnet50 | 0.6892 |
resnet101 | 0.7352 |
python3 inference.py
to test your image by specifying image path and model path there.
python3 train.py --backbone resnet50 --gpu 0 pascal datasets/VOC2012
to start training.python3 utils/eval.py
to evaluate by specifying model path there.