В репозитории код решения соревнования в рамках курса MADE по компьютерному зрению.
Обучение бейзлайна в течение 10 эпох дало 12.7 на паблике :)
Переписал бейзлайн, чтобы использовать imgaug. Поэксперементировал с аугментациями, лучше всего показало себя аффинное преобразование и кроп по центру. Обучил ResNeXt50 - 11.6 на паблике.
Решил попробовать WingLoss c ResNeXt50 - score 10.6.
Захотелось обучить State-of-the-art :) Поискал статьи, наткнулся на HRNet. Взял реализацию на github Допилил голову, чтобы из heatmap'ов получать точки. 10.6 на паблике.
Эксперименты c EfficientNet не дали улучшений.
После всех этих экспериментов, решил сделать флип картинок по горизонтали и обучить ResNet34 c флипом и без него с MSE лоссом. Получил хороший результат на локальной валидации в обоих случаях. Причем без флипа немного лучше. Дотюнил лучший чекпоинт с OneCycleLR scheduler в течение 4 эпох с флипом и без. Лучше получилось с флипом - 9.24 на паблике и 8.97 на private lb.
