做一个分类任务,鱼类识别,
1、数据收集
kaggle有个大规模鱼类数据集 31类一万多张图片
数据集地址: A Large Scale Fish Dataset | Kaggle
爬虫,只能百度图片批量下载
代码:
我删了,网上挺多的
2、模型训练
选择模型
训练保存为.pth权重文件
3、flask本地运行
根据自己的模型准备.pth和.json文件,
运行main.py
得到
就可以在浏览器打开
4、e'cs阿里云服务器部署
创建一个实例
配置安全组,一次性配置好这些可以用到的
下载Xshell 和 Xftp
在Xshell中新建会话
用户名:root
主机:云服务器的公网IP ()
确定
输入账号,密码先在ecs重置实例密码
Xshell连接成功后
点Xftp就可以传输文件了
拖动即可传输文件
重点是在云服务器中安装anaconda和pytorch
4.2 安装conda
参考博客配置服务器:Linux 使用 conda 我所遇到的所有坑 - 掘金
Xshell常用命令大全(附常用实例)_xshell语句-CSDN博客
4.3安装torch、
由于租的云服务器特别小,只有cup,所以只能安装cpu版本的torch和torchversion
查看是否有GPU nnvidia-smi
查看CUDA版本 nvcc --version
下载torch conda install pytorch==1.7.1 cpuonly -c pytorch
网上找的各种方法,哪怕去官网的代码下载也是各种报错,内存不够啥的,上面这句算是不占内存的方法了
5.云服务器xshell中运行
cd /root/fish_mobilenet
ls
#后台运行main.py
nohup python main.py > output.log 2>&1 &
#查找并终止进程
ps aux | grep python
kill -9 <输出的id号>