利用云端GPU训练YOLOv5模型

自己电脑太low了，内存太少了。所以利用下colab的云端GPU来进行训练。
也可以租个服务器来训练，作为新手，我先来白嫖下免费的GPU吧。
在coalb新建一个笔记本，因为coalb中没法直接打开我们的yolov5文件，所以我们需要先将我们的文件压缩打包。在左侧上传我们的压缩文件，我这里是yolov5-6.0.zip

然后我们进行解压
在代码区输入

!unzip /content/yolov5-6.0.zip -d /content/yolov5

/content/yolov5-6.0.zip指的是我们压缩文件的路径

-d 后是我们解压文件的路径

有的时候我们解压后会多出一个文件，_MAXOSX，需要删除下，而且这个玩意没法直接删除，我们需要在代码区输入

!rm -rf /content/yolov5/__MAXOSX

然后我们现在操作是在yolov5这个文件夹下，我们需要进入yolov5中的yolov5-6.0这个文件夹下，在代码区输入

%cd /content/yolov5/yolov5-6.0
# %cd 文件夹路径

然后我们下载我们需要的一些包
一般colab上部分包他都安装好了，我们这个再安装一次。
有的博主是给我们写了requirements.txt文件的，所以我们直接

!pip install -r requirements.txt

添加插件tensorboard(可视化工具)

%load_ext tensorboard

tensorboard --logdir=runs/train

然后我们就可以运行我们的训练文件了

!python train.py

然后我们运行完的模型就保存在了runs/train中

利用云端GPU训练YOLOv5模型相关推荐

TF之pix2pix：基于TF利用Facades数据集训练pix2pix模型、测试并进行生成过程全记录
TF之pix2pix:基于TF利用Facades数据集训练pix2pix模型.测试并进行生成过程全记录目录 TB监控 1.SCALARS 2.IMAGES 3.GRAPHS 4.DISTRIBUTI ...
租用服务器，以及部署在pycharm专业版上的pytorch环境训练yolov5模型教程服务器环境安装库文件：
租用服务器,以及部署在pycharm专业版上的pytorch环境训练yolov5模型教程服务器环境安装库文件: 一.租用服务器选择环境: 1. 租用服务器网站 2. 选择1.8的pytorch环境,点 ...
【华为云技术分享】如何使用pyCharm与ModelArts公有云服务联动开发，快速且充分地利用云端GPU计算资源
其实ModelArts的帮助手册上已经有一篇文档:https://support.huaweicloud.com/bestpractice-modelarts/modelarts_10_0021.ht ...
利用特定领域模型训练Bert模型
目录前言一.预训练模型思路? 二.数据处理 1.MLM任务 2.SOP任务 3.模型搭建总结前言最近刚完成医疗电子病历模型的预训练,经过测试效果还不错,相比于原生Bert在序列标注任务和文本 ...
目标检测（9）—— 使用云端colab训练YOLOV5
colab进行训练将项目打包成zip格式,在colab进行上传:或者上传至谷歌云盘(上传速度更快) 解压缩(我第二种解压方式不知道为什么失败了),这里主要讲解第一种 content就代表我们当前所处 ...
(四)训练用于口罩检测的YOLOv5模型
目录介绍准备训练和验证数据在Colab Notebook上训练YOLOv5模型在Google Colab上测试模型下一步在这里,我们将训练和测试用于口罩检测的YOLOv5模型. 介绍在本 ...
【全流程】从头在树莓派4B上部署自己训练的yolov5模型（配合NCS2加速）
目录 0.前言 1.我的环境 2.整个流程 3.具体过程 3.1 训练自己的yolov5模型 3.2 将.pt模型转换为.onnx模型 3.3 在本地将.onnx转换成IR模型 3.4 在树莓派4B上 ...
Kaldi 使用 DFSMN 训练语音模型
阿里巴巴 2018 年开源的语音识别模型 DFSMN,将全球语音识别准确率纪录提高至 96.04%.DFSMN 模型,是阿里巴巴的高效工业级实现,相对于传统的 LSTM.BLSTM 等声学模型,该模型 ...
keras train_on_batch详解（train_on_batch的输出输入详解，train_on_batch多GPU训练详解，自定义学习率调整策略）
利用 train_on_batch 精细管理训练过程大部分使用 keras 的同学使用 fit() 或者 fit_generator() 进行模型训练, 这两个 api 对于刚接触深度学习的同学非常 ...

利用云端GPU训练YOLOv5模型

利用云端GPU训练YOLOv5模型相关推荐

最新文章

热门文章