/pytorch-lighthead

Light-Head R-CNN in PyTorch

Primary LanguagePythonMIT LicenseMIT

Under development!!!

light-head-rcnn

Developed based on pytorch, backbone replaced with shufflenet v2.
The codes are based on faster-rcnn.
I think it's better to understand the codes of faster-rcnn, if you want to modify the codes(e.g. adding new network)

How to understand the codes

How to generate anchors, Chinese edition

light backbone

I add a state-of-the-art network which is called ShuffleNet v2, because I want to run this detector on mobile phone. If you find some newer and better network, you can add to the project following steps bellow.

How to add new backbone?

  1. In the /lib/model/faster_rcnn/ folder, you can add your network(e.g. EfficientNet, MobileNetv3). You can refer to mobilenet.py to code your own network.
  2. In the /cfgs/ folder, you shoud add a yml file corresponding to your network. Also, you can refer to mobilenet.yml.