Classifier for hair length
pip install -r requirements.txt
python main.py --in-dir "imgs/"
Дополнительно в коде можно настроить так, чтобы bbox для лица и маски волос сохранялись в указанную out-dir
директорию.
Изображение -> bounding box для лица -> маска для волос -> сверточный классификатор на pretrained resnet'e -> метка класса
Есть варианты для модель, которые не опробованы:
- написать свою CNN, которая была бы поменьше;
- попробовать посчитать площадь, координаты, в общем, обойтись без изображений.
Нагенерить в качестве дополнительных данных еще изображений для каждой категории.