Can use pytorch directly train the mtcnn?

[guanfuchen]

first thanks for your sharing code, it is good for me to use pytorch get the model test, I want to know can this model be trained directly using pytorch, instead converting the caffe model to npy.

[TropComplique]

You can't train MTCNN using this code. I haven't implemented training because it's quite burdensome.

I believe that it is quite hard to train MTCNN. And if you want to train your own face detector then consider SSH or FaceBoxes. I think they are more powerful, faster, and easier to train.

[KoapT]

You havent implenmented training? Where are the weights of the 3 nets from?

[minhng92]

He converted the weights from the pretrained Caffe models. Details can be found here:
https://github.com/TropComplique/mtcnn-pytorch/blob/master/extract_weights_from_caffe_models.py

[guanfuchen]

@TropComplique Thanks for your reply, I will try these two models(SSH and FaceBoxes).