search

rainofmine / Face_Attention_Network

Pytorch implementation of face attention network
315 stars 55 forks source link

Usability changes #8

Open ehp opened 5 years ago

ehp commented 5 years ago

Hello,

I made a few changes to your great model to get it more usable:

  1. Codestyle fixes
  2. Proper CPU/CUDA detection
  3. Directories are created at runtime
  4. WIDER dataset loader
  5. ResNet models are downloaded with torch model zoo
  6. Detector script

Let me know if you want something more. E.g. move it into single python package and create proper python library would be nice to have.