Skip to content

Zhangbaisong/DSFP-GA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 

Repository files navigation

DSFP-GA

Discriminative Semantic Feature Pyramid Network with Guided Anchoring for Logo Detection.2021. (in PyTorch)

Description

This repository reproduces "Zhang et al. Discriminative Semantic Feature Pyramid Network with Guided Anchoring for Logo Detection.2021." (DSFP-GA) . The implementation is based on MMDetection framework. All the codes for the DSFP-GA model follow the original paper.

Code

To use this framework, please download it at this link: http://123.57.42.89/Dataset_ict/DSFP-GA.zip

Get Started

To use this repo, please follow README.md of MMDetection.

Train/Test

Train

  • To train baseline (i.e., Faster R-CNN)
python tools/train.py ./configs/dsfp_ga/faster_rcnn_r50_fpn_1x_logo3k.py --work-dir work_dirs/faster_rcnn_r50_fpn_1x_logo3k
  • To train DSFP-GA
python tools/train.py ./configs/dsfp_ga/dsfp_ga_1x_logo3k.py --work-dir work_dirs/dsfp_ga_1x_logo3k

Test

  • To test baseline (i.e., Faster R-CNN)
python tools/test.py ./configs/dsfp_ga/faster_rcnn_r50_fpn_1x_logo3k.py work_dirs/faster_rcnn_r50_fpn_1x_logo3k/faster_rcnn_r50_fpn_1x_logo3k.pth --eval mAP
  • To test DSFP-GA
python tools/test.py ./configs/dsfp_ga/dsfp_ga_1x_logo3k.py work_dirs/dsfp_ga_1x_logo3k/dsfp_ga_1x_logo3k.pth --eval mAP

Benchmark

Below is benchmark results. (extraction code of model : dsfp)

methods backbone mAP download
Faster R-CNN ResNet-50-FPN 83.8 model
DSFP-GA ResNet-50-DSFP 87.7 model

TODO

  • Code reorganization is needed to be consistent with the style of MMDetection framework

About

Discriminative Semantic Feature Pyramid Network with Guided Anchoring for Logo Detection

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published