fa0311/oita-ken-strawberries-mobilenet
By fa0311 · ↓ 0 · ♥ 0 · Updated 2025-02-03
What's in this repo
This page lists the files in fa0311/oita-ken-strawberries-mobilenet, including mb2-ssd-lite-Epoch-55-Loss-1.508891262114048.onnx.
- 1 ONNX models (.onnx files)
- 7 files, including 1 .0, 1 .txt, 1 .onnx, 1 .pth, 1 .md
- Last updated 2025-02-03
ONNX files
| File | Size | Download |
|---|---|---|
| 20250129_053504/mb2-ssd-lite-Epoch-55-Loss-1.508891262114048.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
oita-ken-strawberries-mobilenet
This model is designed for grading the ベリーツ (Beriitsu), a premium strawberry variety cultivated in Oita Prefecture, Japan.
It is a fine-tuned version of MobileNet SSD v2, optimized for strawberry quality classification.
The model has been trained on an augmented dataset derived from oita-ken-strawberries, using image_data_augmentation.
It performs 4-class classification based on an 40,000-image dataset.
Learning
https://github.com/fa0311/pytorch-ssd-forked
python3 train_ssd.py \
--pretrained-ssd=models/mb2-ssd-lite-mp-0_686.pth \
--net=mb2-ssd-lite \
--dataset-type=voc \
--data=data/voc \
--model-dir=models/output \
--num-workers=8 \
--batch-size=16 \
--num-epochs=56 \
--t-0=8 \
--validation-epochs=1 \
--lr=0.0001 \
--extra-layers-lr=0.001 \
--base-net-lr=0.000001 \
--weight-decay=0.000001 \
--validation-mean-ap \
--scheduler=cosine-warmrestart \
--no-augment
Related Resources
Model info
Created: 2025-02-03
Updated: 2025-02-03
View on Hugging FaceFiles (7)
Related models
More from fa0311
No other models by this author.