oita-ken-strawberries-mobilenet

本模型专为日本大分县栽培的高级草莓品种 "ベリーツ"（Beriitsu） 的 分级 而设计。

该模型是 MobileNet SSD v2 的 微调 版本，专门针对草莓品质分类进行了优化。

模型训练使用了基于 oita-ken-strawberries 数据集的 增强 版本，通过 image_data_augmentation 工具处理。

该模型基于 40,000 张图像的数据集，执行 4 类分类 任务。

学习

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

相关资源

• 数据集
• 模型
• 推理演示空间