图像分类
文章平均质量分 74
关于机器视觉中图像分类任务的文章
_-CHEN-_
我很丰富,无法简介
展开
-
部署篇:使用pytorch实现垃圾分类并部署使用,浏览器访问,前端Vue,后台Flask
部署篇:使用pytorch实现垃圾分类并部署使用,浏览器访问,前端Vue,后台Flask1.创建Flask工程2.创建Vue工程1.创建Flask工程打开pycharm,新建工程,选择Flask在工程中新建一个文件夹model用来存放之前训练的模型,从之前的训练工程中把model_best_checkpoint_resnet50.pth.tar和label.txt文件拷贝到model文件夹中,在工程根目录中新建一个python文件,命名为rubbish.py,这个文件用来调用模型进行预测。代码如原创 2021-11-14 16:13:17 · 2317 阅读 · 4 评论 -
训练篇:使用pytorch实现垃圾分类并部署使用,浏览器访问,前端Vue,后台Flask
训练篇:使用pytorch实现垃圾分类并部署使用,浏览器访问,前端Vue,后台Flask1.数据集准备1.1数据集下载1.2 数据集划分1.3实现自己的Dataset2.开始训练数据2.1 实现训练方法和测试方法2.2 开始训练1.数据集准备1.1数据集下载在百度的AI Studio中有公开数据集,里面有很多不错的数据集可以给我们免费下载使用,下载速度也很快其中这个数据集就是本文所使用的的数据集,下载链接如下https://aistudio.baidu.com/aistudio/datasetd原创 2021-11-13 10:52:40 · 1747 阅读 · 0 评论