search

chenjun2hao / CenterFace.pytorch

unofficial version of centerface, which achieves the best balance between speed and accuracy at face detection
236 stars 60 forks source link

Accuracy Issue #40

Open PhenomenalOnee opened 3 years ago

PhenomenalOnee commented 3 years ago

Really Nice Tutorial.. I have trained using code from https://github.com/siriusdemon/Build-Your-Own-Face-Model/tree/master/detection centerface detection and facing Accuracy Issue. Some Modifications I have done according to your repo as follows HeatMap generation using your code. FocalLoss implementation for heatmap loss removed landmark generation layer. 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?

PhenomenalOnee commented 3 years ago

@Watebear yes sure we can . I have mailed you. we can contact and start this journey