search

MarekKowalski / DeepAlignmentNetwork

A deep neural network for face alignment
MIT License
509 stars 138 forks source link

how to train model via the codes? #19

Open DcYond opened 6 years ago

DcYond commented 6 years ago

@MarekKowalski Thanks for sharing the codes. how to train model via the codes?

MarekKowalski commented 6 years ago

You should use the DANtraining.py script. Before you do that however you need to "build" the training set by running: TrainingSetPreparation.py You will also need to download the 300-W training set from the IBUG website beforehand.

Marek

hengshan123 commented 6 years ago

Hi I want to know how many epochs should each stage train? I see the num_epochs is 1000 in DANtraining.py, this just means the training is run in 1000 * 64 images?

So both of the two stages is trained by 1000 epoch? is that right?

MarekKowalski commented 6 years ago

Hi,

The training is ran for as long as the validation error is decreasing "fairly rapidly". I would finish training if a lower error is not achieved after, let's say two or three hours since the last best error was achieved. The time and number of epochs this takes varies.

Marek