用yolov5在colab上训练自己的数据集

最近的课题是用yolov5进行目标检测，在此写个笔记记录一下自己的学习过程。
方便以后回顾和监督自己。
由于学校服务器尚未申请下来，故在colab上进行了实验。
colab每次断开连接以后都要重头搭配环境，也算省了一个创建虚拟环境的过程😂。
yolov5 代码连接：https://github.com/ultralytics/yolov5

配置环境

在google云端硬盘中创建一个单独的文件夹，将数据集放在里面，创建一个notebook。
1，创建好notebook，修改–>笔记本设置 设置GPU加速
2，连接–>选择连接到托管代码执行程序
3，查看提供的GPU资源

!nvidia-smi

4，查看pytorch的安装情况

import torch
torch.__version__

5，挂载google drive

import os
from google.colab import drive
drive.mount('/content/drive')
 
path = "/content/drive/My Drive"
 
os.chdir(path)
os.listdir(path)

6，下载yolov5源码

!git clone https://github.com/ultralytics/yolov5

7，安装所需的环境

#进入yolov5文件夹
%cd yolov5 
#安装
%pip