search

yeephycho / tensorflow-face-detection

A mobilenet SSD based face detector, powered by tensorflow object detection api, trained by WIDERFACE dataset.
Apache License 2.0
768 stars 264 forks source link

tensorflow model #23

Closed engr77 closed 6 years ago

engr77 commented 6 years ago

Hi, can you please share your tensorflow model? Your model is quite unique that it can accept non 1:1 aspect ratio images. Mobilenet can only take 1:1 ratio model. Wondering how you designed your model and trained it. Really appreciate if you can share!

RobvanGastel commented 6 years ago

I'm also really interested how the training script looks like!

yeephycho commented 6 years ago

Hi @engr77 , I think there's resize operation before feed the image into the model, so, aspect ratio is not a problem. And @RobvanGastel , to be frankly, the training process is to long and you guys can find the training tutorial through tensorflow object detection api. I also already forgot how did I train this model. And the widerface_to_tfrecord repository maybe helpful for you guys https://github.com/yeephycho/widerface-to-tfrecord

RobvanGastel commented 6 years ago

Alright thanks for the reference @yeephycho