- #
- 1
- idstring
- 0
Загрузка
Загружаем соревнования
К задаче прикреплены файлы:
train.csv — размеченных клеток: колонки id, label;val.csv — размеченных клеток: колонки id, label;test.csv — Датасет ограничен
CSV66 Б5 строк1 колонок
| # | idstring |
|---|---|
| 1 | 0 |
| 2 | 1 |
| 3 | 2 |
| 4 | 3 |
| 5 | 4 |
Первые 5 строк · только просмотр
iddata/train/<id>.jpg, data/val/<id>.jpg, data/test/<id>.jpg — сами клетки, RGB ;classes.csv — соответствие label → название класса: 0 Forest, 1 Highway, 2 Industrial, 3 AnnualCrop, 4 SeaLake;weights/mobilenet_v2-b0353104.pth — веса MobileNetV2 (ImageNet), state_dict для torchvision.models.mobilenet_v2;homework_pevtsov.ipynb — ноутбук с заданием;sample_submission.csv — пример решения, которое нужно отправить в тестирующую систему.В train по клеток каждого класса, в val по , в test классы сбалансированы. Клетки нового участка сняты с произвольной ориентацией.
Загрузка модели без сети:
import torch
from torchvision.models import mobilenet_v2
model = mobilenet_v2(weights=None)
model.load_state_dict(torch.load("weights/mobilenet_v2-b0353104.pth", map_location="cpu"))