使用GPU云平台给V831训练模型

我用的是恒源云。（新人有个充三十送三十）

这是恒源云官网：恒源云_GPUSHARE-恒源智享云

首先注册，（好像新人还可以免费用一下的，记不清了，如果没有送就砸钱，打电赛砸三十就够了）注册完点控制台。

 然后创建实例

 因为充三十送的三十是代金卷，所以记得把代金卷选项勾上，然后可以选择要租用的显卡型号

 我自己成功的试例用的是3090，然后选择官方镜像

 我们根据官方文档所要求的版本选择

 版本只能高不能低（但是建议不要选太新的版本，因为新版本会淘汰一些东西，导致报一些错误）我选择的是这个版本

 建立了实例后，我们还需要使用FileZilla来上传文件，用Xshell 7 来链接服务器，需要下载注册，Xshell和FileZilla的教程恒源云上有

 

 可自行阅读。

然后用Xshell链接服务器后，出现下面界面

 然后输入这段代码，下载所需要的库。

pip3 install torchsummary pycocotools opencv-python opencv-contrib-python

下完可能会报一下错误，暂时不管它，到最后跑模型如果报错，再去搜。

接下来我们先在官网下载模板文件

这是链接：https://github.com/dianjixz/v831_yolo

然后制作数据集

 根据模板文件，把数据替换为自己的拍的数据

制作完后，压缩为压缩包，然后通过FileZilla传到服务器的 /hy-tmp 位置

传完后在Xshell里输入

cd /hy-tmp          (进入hy-tmp文件夹)

ls                        (查看当前文件夹里的文件)

这时你就可以看到你传进去的压缩包

然后输入

unzip 压缩包名称.zip

解压完这个时候可能有两种情况，你输入ls查看，如果是解压出来一个文件夹，你就

cd 文件夹名称    （进入文件夹）

然后输入训练模型指令

python3 train.py -d custom --cuda -v slim_yolo_v2 -hr -ms

如果你输入ls是一堆解压出来的文件，你就直接输入训练模型指令。（当然还是建议大家在压缩文件时在外面包一层文件夹再压缩，毕竟文件散落在hy-tmp文件夹里不好）

然后就开始漫长的跑模型了，如果出现了什么错误打断了跑模型，就复制错误然后寻找解决方法。

比如这一步里面的标签只能为小写（我也不知道为啥）

 等它跑完就导出模型

python3 test.py -d custom -v slim_yolo_v2 --trained_model weights/custom/slim_yolo_v2/slim_yolo_v2_260.pth --visual_threshold 0.3 -size 224 --export

运行这串代码，它会在out文件夹生成一个.onnx文件，然后我们用FileZilla将文件导出，去MaixHub转换。

MaixHub网址：MaixHub

 

 要上传一个包含你素材和onnx文件的压缩包（素材就你每个目标上传个十几二十张）

只要上传压缩包就行，它那些参数不用去改它的

最后在左下角下载最后的文件，烧进v831就可以用了。

 具体模型部署参考官网：目标检测本地训练教程 - Sipeed Wiki

 希望能给你提供到帮助