colab进行训练
- 将项目打包成zip格式,在colab进行上传;或者上传至谷歌云盘(上传速度更快)
- 解压缩(我第二种解压方式不知道为什么失败了),这里主要讲解第一种
content就代表我们当前所处的文件夹,-d后面表示要压缩在哪里,还是压缩至当前的文件夹。
- 进入到项目的根目录
- 配置环境
- 配置tensorboard
- 开始训练
- 训练了很多轮后刷新一下tensorboard,有一些变化的
训练的目录结构:
只看当前的:
指定使用整个coco数据集:
8. 训练好可以将最好的模型下下来:
- 下好后放在根目录,在detect.py中把weights参数改成你自己的模型就OK!