PlantNet-300K — ResNet18 (45 MB)

Fine-tuned ResNet18 for plant species identification across 1,081 species.

This model serves as the reference / upper-bound in an experiment on small, locally-deployable plant classifiers. It is compared against a much lighter MobileNetV3-Small (10 MB) to assess the accuracy cost of radical size reduction.

👉 Live demo: cpoisson/plantnet300k
👉 Lightweight model (MobileNetV3-Small, 10 MB): cpoisson/plantnet300k-mobilenetv3-small

---

Model Details

---

Dataset: Pl@ntNet-300K

Garcin et al., "Pl@ntNet-300K: a plant image dataset with high label ambiguity and a long-tailed distribution", NeurIPS 2021.

---

Training

Training script: train.py (included in this repo).

Hardware

---

Results — Test Set (31,112 images)

---

Model Comparison

ResNet18 gains +1.93 pp top-1 at 4.5× the size. Whether that trade-off is worth it depends heavily on the deployment target.

---

How to Replicate

1. Download the dataset

wget https://zenodo.org/records/5645731/files/plantnet_300K_images.tar.gz
tar -xzf plantnet_300K_images.tar.gz

2. Train

# Edit DATA_DIR and set model = models.resnet18(...) in train.py
python train.py

3. Load & infer

import torch
from torchvision import models, transforms
from huggingface_hub import hf_hub_download
from PIL import Image

model = models.resnet18(weights=None, num_classes=1081)
path = hf_hub_download("cpoisson/plantnet300k-resnet18", "plantnet_resnet18.pth")
model.load_state_dict(torch.load(path, map_location="cpu", weights_only=True))
model.eval()

transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])

img = Image.open("your_plant.jpg").convert("RGB")
with torch.no_grad():
    logits = model(transform(img).unsqueeze(0))
    probs  = torch.softmax(logits, dim=1)[0]
    top5   = probs.topk(5)

---

Citation

@inproceedings{plantnet-300k,
  author    = {Garcin, Camille and Joly, Alexis and Bonnet, Pierre and Lombardo, Jean-Christophe
               and Affouard, Antoine and Chouet, Mathias and Servajean, Maximilien
               and Lorieul, Titouan and Salmon, Joseph},
  booktitle = {NeurIPS Datasets and Benchmarks 2021},
  title     = {{Pl@ntNet-300K}: a plant image dataset with high label ambiguity and a long-tailed distribution},
  year      = {2021},
}