GZSL Weeds Identification: Lightweight Classifier Weights

This repository hosts the PyTorch checkpoints used in our generalized zero‑shot learning (GZSL) pipeline for weed identification in agricultural imagery.
Backbones were fine‑tuned on CropAndWeed and evaluated for cross‑dataset generalization to Plant Phenotyping and a self‑collected, real‑field dataset.

Available models

File name Architecture / variant
mobilenet.pt MobileNetV2 (ImageNet stem, width 1.0)
resnet18.pt ResNet‑18
squeezenet.pt SqueezeNet 1.1
shufflenet.pt ShuffleNet V2 (baseline)
shufflenet_squeeze_excitation.pt ShuffleNet V2 + Squeeze‑and‑Excitation (SE)
shufflenet_sep_conv.pt ShuffleNet V2 + Depthwise Separable Convolution (SC)
shufflenet_sep_conv_squeeze_excitation.pt ShuffleNet V2 + SC + SE

Getting started

All inference scripts, data loaders and architecture definitions live in the companion GitHub repository: https://github.com/SyArsRa/WeedZSL.git

The quick‑start guide there walks through:

  1. Instantiating the desired backbone (for example, MobileNetV2 or ShuffleNetV2 + SE)
  2. Loading the matching .pt file from this weights hub
  3. Running single‑image or batch inference
  4. Fine‑tuning on a custom dataset if needed

License

Weights and code are released under the MIT license for research and non‑commercial use.
See LICENSE for details or contact the maintainers for alternative licensing.

Citation

These checkpoints support a study currently submitted to a conference.
Please cite the forthcoming paper or contact the authors for an interim reference.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support