search

TropComplique / mtcnn-pytorch

Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks
MIT License
659 stars 161 forks source link

Can use pytorch directly train the mtcnn? #3

Open guanfuchen opened 6 years ago

guanfuchen commented 6 years ago

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 commented 6 years ago

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 commented 6 years ago

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

minhng92 commented 6 years ago

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 commented 6 years ago

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