Skip to content

Latest commit

 

History

History
37 lines (33 loc) · 1.06 KB

README.md

File metadata and controls

37 lines (33 loc) · 1.06 KB
Raw

LFFD: A Light and Fast Face Detector for Edge Devices

paper link

Requirements:

  • pytorch>=1.2
  • opencv-python
  • torchvision>=0.4

Label Assignment

Boxes are in different scale. Anchor points with more than one matched gt box are ignored.
green box --> gt box
blue/red point --> positive sample
The positive sample sampling of the fifth branch is shown below.

Different settings from the paper

  • regression target
  • add BN
  • more classes

Test

I have trained the model on a safety helmet dataset without any augmentation. It seems to be working well.

tensorboard

test_result