search

StanislasBertrand / RetinaFace-tf2

RetinaFace (RetinaFace: Single-stage Dense Face Localisation in the Wild, published in 2019) reimplemented in Tensorflow 2.0, with pretrained weights available !
MIT License
258 stars 21 forks source link

Could you add a training file? #6

Closed wulikai1993 closed 4 years ago

wulikai1993 commented 4 years ago

And how long time it takes to train a model? Do you train the model on a single GPU or multi GPUs?

StanislasBertrand commented 4 years ago

Hi,

There is no training method provided. I converted the already trained weights and code from https://github.com/deepinsight/insightface/tree/master/RetinaFace

Although the provided model is a classif tf.keras model, you could try to train it the usual tf.keras way. In order to match the articles method you would probably also need to implement / find a tf.keras implementation of their multi task loss