- В папке utils лежат используемые утилиты для визуализации и подсчета accuracy для каждого класса
- В папке models_viz лежат визуализации используемых архитектур CNN
- custom_cnn - файл с архитектурой кастомной CNN
- training - главный файл с полным пайплайном обучения и инференса двух моделей