Решить задачу CV по распознаванию фигур

Description

В распоряжении есть набор изображений, которые состоят из нескольких видов геометрических фигур:

• треугольников;
• пентагонов;
• ромбов;
• окружностей.

Задача – посчитать количество треугольников на каждом изображении из тестовых данных.

Metric

В качестве метрики качества используется метрика Root Mean Square Error (RMSE)

Submit

Сабмит должен содержать две колонки:

• ID – идентификатор записи
• target – число треугольников на изображении

Общие сведения

Весь набор данных состоит из 3 типов изображений:

• Содержат только треугольники, которые могут иметь общие стороны, вкладываться друг в друга, но не пересекаться
• Содержат треугольники и другие фигуры, все они могут иметь общие стороны, вкладываться друг в друга, но не пересекаться
• Содержат треугольники и другие фигуры, которые дополнительно могут пересекаться
  Фотографии для обучения и теста находятся в папках train_data и test_data соответственно.

Есть 3 основных файла:

• train_dataset.csv – содержит информацию об обучающих изображениях
• test_dataset.csv – содержит информацию о тестовых
• test_submission.csv – пример загружаемого ответа

train_dataset.csv

train_dataset.csv имеет следующие поля:

• path – путь до изображения
• num_shapes – число треугольников на изображении (только в обучающем наборе)

test_dataset.csv

test_dataset.csv имеет следующие поля:

• path – путь до изображения
• ID – идентификатор записи