一、首先开启魔法
这里就不展示了。
二、打开网站
https://colab.google/
点击open colab。
点击修改中的笔记本设置,选择GPU加速。
因为colab无法上传文件夹,首先需要将自己的代码压缩一下。
1、点击文件 2、点击上传刚刚压缩好的代码。
三、模型的训练
首先,点击左上角的+代码,添加一行命令用于将压缩的文件解压。-d参数用于确定解压后的代码的位置。上图所示的文件所在位置默认为/content/。
!unzip /content/yolov5-6.0.zip -d /content/yolov5
再切换到当前目录
%cd /content/yolov5/yolov5-6.0
根据代码中的requirements.txt下载需要的依赖库
!pip install -r requirements.txt
可以使用tensorboard进行可视化(可选)
%load_ext tensorboard
%tensorboard --logdir=runs/train
最后,运行代码
!python train.py