Plant Disease CNN

Небольшая CNN-модель от Media1234 предсказывает по фотографии листа один из 39 классов PlantVillage: конкретное растение и болезнь или здоровый лист. Это учебный проект для KCL 5CCSAGAP, но с готовым чекпоинтом, кодом инференса и описанием ограничений.

👀 Ключевые особенности

• 👀 CNN: num_classes=39, base_channels=32, depth=4, dropout=0.3, BatchNorm.
• 🍒 Вход — RGB 128×128 с нормализацией ImageNet mean/std.
• 💪 Лучшая validation accuracy — 0.9898.
• 🐾 Обучена на стратифицированном 70% сплите DScomp380/plant_village.
• 😐 Знает только 39 классов: другие растения и болезни будут отнесены к одному из них.

👉 Пример

from huggingface_hub import hf_hub_download
from src.inference import Predictor

predictor = Predictor(hf_hub_download("Media1234/plant-disease-cnn", "cnn.pt"))
print(predictor.top_k(pil_image, k=5))

Подробнее

Validation accuracy: 0.9898 на 39 классах.
PlantVillage — один лист на простом фоне; на полевых фото с несколькими листьями и другим светом точность будет ниже.
Это студенческий проект, а не диагностический инструмент.
Формат чекпоинта — cnn.pt, PyTorch.