训练自己的yolo-tiny模型

  在文章 训练自己的yolov3模型中介绍了如何训练自己的模型,但是此时可能会出现一种情况就是自己的GPU性能不太好或者是使用的CPU进行的训练,那么这里有另外一种选择进行训练,训练yolov3-tiny,这是一种小型的网络,不管是训练还是移植,速度都是比较快的,它相比于yolov3用以牺牲精度换来速度的提升。当然,如果你一定要训练yolov3模型,或者想要在服务器上跑yolov3-tiny模型,可以将训练文件放到服务器上跑,如何在服务器上跑自己的模型在文章链接: 在服务器上训练自己的yolo模型.有详细的介绍,想要用服务器跑的可以移步。这里介绍一下yolov3-tiny的训练与yolov3训练有何异同
  在yolov3-tiny的训练过程中,其前6步与 训练自己的yolov3模型中的步骤相同
  在第七步中,修改的cfg文件是yolov3-tiny.cfg,还需注意的是,在这个cfg文件中搜索yolo只有两个结果,只需要修改两次那三个参数即可
  在第8步中,需要下载预训练权重,这个权重不是直接下载的,需要先下载yolov3-tiny.weights再从中抽取,具体的操作方法如下

wget https://pjreddie.com/media/files/yolov3_tiny.weights  #下载预训练权重
./darknet partial ./cfg/yolov3-tiny.cfg ./yolov3-tiny.weights ./yolov3-tiny.conv.15 15   #从下载的权重中抽取预训练需要的

  第9-11步还是相同的操作与yolov3的训练相同,这里不再赘述

你可能感兴趣的:(yolov3训练,深度学习,人工智能,计算机视觉,神经网络)