利用迁移学习进行花的分类

前几天写了个小项目, 利用深度学习里的迁移学习方法做花的分类, 下面是项目的github地址.
https://github.com/JameyWoo/transfer-learning

感兴趣的同学可以了解一下, 源码比较简单, 下面是项目的说明

transfer learning

@ author 姬小野

— 迁移学习对五种花分类

环境

ubuntu 18.04

requirements

torch==1.1.0numpy==1.17.0torchvision==0.3.0

使用方法

训练

下载vgg的预训练模型
download.pytorch.org/models/vgg16-397923af.pth放到目录/home/jamey/.cache/torch/checkpoints下
执行
python train.py

即可在当前目录下训练自己的模型

ps. 在普通笔记本上生成模型的时间较久

测试

执行

python test.py

即可测试模型对花分类的准确率

当epoch为3时, 模型的准确率达到了83%, 其中, 除roses外准确率都极高. (大多数错误都是由roses引起的)

Test Accuracy of daisy: 84% (78/92) Test Accuracy of dandelion: 94% (125/132) Test Accuracy of roses: 59% (54/91) Test Accuracy of sunflowers: 85% (86/101) Test Accuracy of tulips: 87% (108/124) Test Accuracy (Overall): 83% (451/540) 1234567运行

对一张图片进行分类

----------------usage---------------- run the demo with: python demo.py -m model_name -i image_name.jpg python demo.py --image image_name.jpg python demo.py -i image_name.jpg or use `python demo.py -h` to get help -----------------end----------------- 1234567

例子

demo 1

$ python demo.py -m my_vgg16_3epochs.pth -i image/yvjingxiang_1.png 郁金香 12

demo.py 的输出结果是图片的识别的花的中文名

demo 2

$ python demo.py -m my_vgg16_3epochs.pth -i image/pugongying_1.png 蒲公英 12

相关知识

使用迁移学习对花卉进行分类
基于深度学习和迁移学习的识花实践
基于迁移学习的苹果树叶片病虫害识别方法研究
基于深度迁移学习模型的花卉种类识别
智能识别花生病虫害：应用迁移学习与CNN
迁移学习常见蔬菜害虫的检测分析
基于迁移学习的花卉识别系统
基于残差网络迁移学习的花卉识别系统
【pytorch】resNet152迁移学习实现植物病害图像识别分类
深度学习花的分类识别

原文链接: 利用迁移学习进行花的分类 https://www.huajiangbk.com/newsview516092.html

上一篇: 猫眼石鉴别:3招辨真伪 – 根盆...

下一篇: 花卉分类和识别习题.ppt