Home

Awesome

FSAF

This is an implementation of FSAF on keras and Tensorflow. The project is based on fizyr/keras-retinanet and fsaf branch of zccstig/mmdetection. Thanks for their hard work.

As the authors write, FASF module can be plugged into any single-shot detectors with FPN-like structure smoothly. I have also tried on yolo3. Anchor-free yolo3(with FSAF) gets a comparable performance with the anchor-based counterpart. But you don't need to pre-compute the anchor sizes any more. And it is much better and faster than the one based on retinanet.

Updates

Test

  1. I trained on Pascal VOC2012 trainval.txt + Pascal VOC2007 train.txt, and validated on Pascal VOC2007 val.txt. There are 14041 images for training and 2510 images for validation.
  2. The best evaluation results (score_threshold=0.05) on VOC2007 test are:
backbonemAP<sub>50</sub>
resnet500.7248
resnet1010.7652
  1. Pretrained models are here.
    baidu netdisk extract code: rbrr
    goole dirver

  2. python3 inference.py to test your image by specifying image path and model path there.

image1 image2 image3

Train

build dataset (Pascal VOC, other types please refer to fizyr/keras-retinanet)

train

Evaluate