使用colab训练yolov3模型

关于colab的使用,请自行参考网上相关的教程,本处不重复描述

  1. 下载ipynb文件

  2. 将其上传至colab

  3. 依据提示执行使用colab训练yolov3模型_第1张图片

  4. Upload dataset这步时,跳过这一步,并上传你的含有训练数据的data.zip文件
    使用colab训练yolov3模型_第2张图片修改此处代码为压缩包路径 使用colab训练yolov3模型_第3张图片解压后确保文件名为data,不然之后步骤需要修改代码使用colab训练yolov3模型_第4张图片

  5. 依照步骤执行,执行完 Create config files for training 后,将会生成三个文件在这里插入图片描述这一步需要注意,务必不要重复执行代码,执行后生成的文件不是覆盖原文件,而是在原文件后面继续写入。所以如果需要修改内容,务必先删除对应文件,再修改代码并执行

  6. 修改 train.sh 文件,将此目录改为当前路径
    使用colab训练yolov3模型_第5张图片使用colab训练yolov3模型_第6张图片

  7. 然后就可开始训练,训练完之后,将-yolov3-tiny.cfgbackup文件夹中的-yolov3-tiny_final.weights下载下来即可移植至其他设备做推理

你可能感兴趣的:(使用colab训练yolov3模型)