Для этой задачи на тренировочном наборе данных выполнялись следующие аугментации:
- VerticalFlip, HorizontalFlip и Rotate: Это может быть полезно, поскольку кошки и собаки могут быть сфотографированы с разных ракурсов и углов. Кошка вверх ногами всеравно кошка.
- CLAHE (Contrast Limited Adaptive Histogram Equalization): Эта аугментация улучшает контраст изображения, тем самым лучше распознаются объекты на изображении.
- RandomBrightnessContrast: Случайным образом изменяет яркость и контраст изображения. Позволяет модели работать с данными при различных условиях освещения.
- RandomGamma: Изменяет гамму изображения, что также может помочь модели лучше справляться с различными условиями освещения.
- Resize: Меняет размер изображения. (в данном случае, 224x224). Это необходимо, поскольку модель имеет фиксированый входной слой.
- Normalize: Нормализует значения пикселей изображения.
Для тестового и валидационных наборов данных выполнялись только 5. и 6.