В репозитории:
- Main_notebook.ipynb - Главный ноутбук с обучением модели на 500 классов, не дающими качественный результат наработками Triplt loss и ArcFace loss, работающим в силу обученности внутренних моделей пайплайн Face Recognition(classification).
- Alignment_training.ipynb - Ноутбук с обучением модели нахождения лэндмарков на изображении лица.
- Prepare_data_for_YoloV5.ipynb - Ноутбук подготовки разметки для Yolov5
- utils.py - Функции и классы хелперы.
- trained_models - Папка с моделями, использующимися в главном ноутбуке
- Папка Best_accscore_1000 с ноутбуком обучения модели на CE на 1000 классов с accuracy > 0.85 на валидации.
TODO: Robust landmarks rotation
TODO: Retrain Alignment NN with more custom random augmentations
TODO: ArcFace