Iris landmarks localization.
Repo for dense iris landmarks localization with synthesized eye dataset.
./config.py: config file
./loss.py: loss function
./checkpoint.py: save the trained model
./tools: some utitilies
./test: test results
There are two methods to prepare the training data.
You could use the software here to synthesiz all kinds of data yourself. Then use scripts in ./gen_dataset
to generate training data.
You could also use the dataset I provided. directly. Just download the dataset and put train images in ./data
directory.
In this case the annotations are already prepared in annotations
directory.
Adress:https://pan.baidu.com/s/1gzYAVvEuhuu6L8tos3zXAQ
Password:990n
config all the training parameters in config.py
RUN python training/train.py
to train your model.
Training results are kept in results
directory.
Put well croped eye images in ./data/test
RUN python test/test_image.py
to test your model.