利用迁移学习进行花的分类
前几天写了个小项目, 利用深度学习里的迁移学习方法做花的分类, 下面是项目的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://m.huajiangbk.com/newsview516092.html
| 上一篇: 猫眼石鉴别:3招辨真伪 – 根盆 |
下一篇: 花卉分类和识别习题.ppt |
