diff --git a/README.md b/README.md index 6245196..08e03cb 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # Attention-guided Context Feature Pyramid Network for Object Detection -This repository is written by Junxu Cao at UC San Diego, on the base of [Detectron](https://github.com/facebookresearch/Detectron) @ [e8942c8](https://github.com/facebookresearch/Detectron/tree/e8942c882abf6e28fe68a626ec55028c9bdfe1cf). +This repository is written by Junxu Cao, on the base of [Detectron](https://github.com/facebookresearch/Detectron) @ [e8942c8](https://github.com/facebookresearch/Detectron/tree/e8942c882abf6e28fe68a626ec55028c9bdfe1cf). ## Introduction @@ -8,6 +8,75 @@ This repository re-implements AC-FPN on the base of [Detectron](https://github.c Please follow [Detectron](https://github.com/facebookresearch/Detectron) on how to install and use Detectron-Cascade-RCNN. +## AC-FPN + +Because of the proposed architecture, We have better performance on bigger objectWe have better performance on bigger object +![architecture](pics/architecture.jpg) + +Object detection result +![detection](pics/detection_samples.png) + +Instance segmentation result +![segmentation](pics/instance_samples.png) + +More detail in paper. + +## Benchmarking + +AC-FPN can be readily plugged into existing FPN-based models and improve performance. +![segmentation](pics/paper_result.png) + +This repo has released CEM module without AM module, but we can get higher performance than the implementation of pytorch in paper. +Also, thanks to the power of detectron, this repo is faster in training and inference. + +The result of coco test-dev(team Neptune) +![rank](pics/rank.png) +### Mask R-CNN with Bells & Whistles + +
backbone | +type | +lr schd |
+im/ gpu |
+box AP |
+box AP50 |
+box AP75 |
+
+
---|---|---|---|---|---|---|
X-152-32x8d-FPN-IN5k-baseline | +Mask | +s1x | +1 | +48.1 | +68.3 | +52.9 | +
X-152-32x8d-FPN-IN5k-cascade | +Mask | +s1x | +1 | +50.2 | +68.2 | +55.0 | +
X-152-32x8d-FPN-IN5k-acfpn(just CEM) | +Mask | +s1x | +1 | +51.9 | +70.4 | +57.0 | +