search

Sierkinhane / mtcnn-pytorch

A face detection algorithm
317 stars 162 forks source link

Onet trained model not working #4

Open soham24 opened 5 years ago

soham24 commented 5 years ago

I have trained onet by following exact same steps on the same dataset. But when I have replaced it with onet from the demo example. it is not working.

Z-Jeff commented 4 years ago

I have meet the same problem. P-Net and R-Net work well, but O-Net doesn't detect any faces or landmarks. Have you solve this problem?

soham24 commented 4 years ago

@Z-Jeff I did not debug it further. You can check blazeface pytorch implementation if looking for face detector

Z-Jeff commented 4 years ago

I have solve this problem just now. 'boxes' should be the form of [x1, y1, x2, y2], while the data in "./anno_store/anno_train.txt" is [x1, x2, dx, dy]. I have changed the format in 'gen_Pnet_train_data.py' and 'gen_Rnet_train_data.py', but forgotten to modify 'gen_Onet_train_data.py'. After changing the format in 'gen_Onet_train_data.py', the O-Net works.