search

siriusdemon / Build-Your-Own-Face-Model

Build your own ArcFace, CenterFace/Centernet
332 stars 69 forks source link

Training Issue #13

Closed PhenomenalOnee closed 3 years ago

PhenomenalOnee commented 3 years ago

Really Nice Tutorial.. I have trained using your code for centerface detection and facing Accuracy Issue. Some Modifications I have done according to https://github.com/chenjun2hao/CenterFace.pytorch 1) HeatMap generation using code from mentioned repo. 2) FocalLoss implementation for heatmap loss 3) removed landmark generation layer. 4) Data augmentation ( random crop, random scale, translation) from https://github.com/Paperspace/DataAugmentationForObjectDetection

I have trained Mobilnet_10 model with pretrained weights upto 90 epochs with LR 5e-4 and facing accuracy problem as compared to original centerface model. Also during inference if i use threshold to 0.50, 1000+ face detections are made (false detection) Can you help?

siriusdemon commented 3 years ago

Refer to DBface. It may be helpful.