关于colab的使用,请自行参考网上相关的教程,本处不重复描述
-
下载ipynb文件
-
将其上传至colab
-
依据提示执行
-
到
Upload dataset这步时,跳过这一步,并上传你的含有训练数据的data.zip文件修改此处代码为压缩包路径
解压后确保文件名为data,不然之后步骤需要修改代码
-
依照步骤执行,执行完
Create config files for training后,将会生成三个文件这一步需要注意,务必不要重复执行代码,执行后生成的文件不是覆盖原文件,而是在原文件后面继续写入。所以如果需要修改内容,务必先删除对应文件,再修改代码并执行
-
修改
train.sh文件,将此目录改为当前路径 -
然后就可开始训练,训练完之后,将
-yolov3-tiny.cfg和backup文件夹中的-yolov3-tiny_final.weights下载下来即可移植至其他设备做推理