Дататон МТМ 2021. Задача №5. Классификация животных (уровень сложности – начинающий)

Финиш 29 апреля 23:59

Поздравляем с прохождением в продвинутый этап Дататона! Это первая задача, уровень сложности – начинающий.

Задача «Классификация животного»

Создайте классификатор, который будет отличать изображения кошек от изображений собак.

Система классификации

Подготовлена программа для предварительной обработки изображений, проектирования и настройки нейронной сети. Необходимо отредактировать эту программу для достижения наилучшего результата. Язык программирования: Python

Подготовка

Посмотрите вводный онлайн-курс по нейронным сетям на сайте Академии Высоких Технологий: “Нейронные сети для самых маленьких”.

Выполнение

Скачайте архив с данными. В нём находятся файлы main.py, eval.py и две папки с животными: animals (тренировочный набор данных) и animalsval (валидационный).

В файле eval.py необходимо заполнить 3 функции:

  1. Функцию предобработки данных
  2. Фунцию загрузки вашей модели
  3. Функцию предсказания класса животного на картинке: 0 - кошка, 1 - собака.

Не меняйте названия функций файла eval.py.

Всю работу по отработке модели необходимо вести в файле eval.py. Файл main.py предназначен для проверки работоспособности и точности модели на валидационной выборке. Как только точность Вас удовлетворит, вернитесь на эту страницу и загрузите файл eval.py, нажав на кнопку “прислать решение”.

Проверка кода осуществляется на тестовых данных: изображениях, не включенных в тренировочную и валидационную выборки. Метрикой оценки является F1-мера.

В качестве решения присылайте архив с файлом eval.py и с сохранённой моделью в формате .h5 (Как сохранять модель в формат h5). Архив должен называться animal_task.zip

Важно:

  1. Для построения модели нейронной сети используйте tensorflow версии 2.0.1. Если вы используете Google Colab, то установить нужную версию tensorflow можно следующим образом, это нужно делать каждую сессию: TF
  2. Архив с заданием должен называться animal_task.zip и содержать файлы eval.py и вашу модель



Очки Участник Дата Команда Город Учебное заведение
0.981 Shaman 28.04.2021 17:59
0.94 Anton3004 27.04.2021 18:32
0.915 Олег Кочетков 27.04.2021 13:43 Тестирование
0.679 Виктор Астафьев 27.04.2021 18:18 Robodriver Москва Международная школа завтрашнего дня (IST)
0.667 Василий Белкин 21.04.2021 14:14 Москва