search

lzx1413 / PytorchSSD

pytorch version of SSD and it's enhanced methods such as RFBSSD,FSSD and RefineDet
MIT License
709 stars 237 forks source link

Add HarDNet as backbone models #93

Closed PingoLH closed 5 years ago

PingoLH commented 5 years ago

Hi lzx, I'd like to contribute four SSD/RFBNet models with HarDNet68/85 as backbone models to this repo. Please note that:

  1. utils/box_utils.py: I believe there is a bug with the fully reduced dim for the sum
  2. I also made some modifications for supporting pytorch 0.4.1 ~ 1.1.0. I've tested it with 0.4.1, 1.0.1, and 1.1.0. However, I couldn't test it with the earlier versions due to the cuda version, so I'm not sure if it will conflict to 0.3.1 and earlier versions.
  3. if there is any concern with the modification on train_test.py, we can rename it to train_test_hardnet.py Thank you so much!