第六章 YOLO v2

简介

该代码主要参考了longcw/yolo2-pytorch的PyTorch复现工程，如在学习时遇到问题，可前往longcw的问题区查看是否有解决方法。

准备工作

1 编译

• 编译 reorg 模块，修改mask.sh中的arch，具体可参考第四章。

  cd yolo2-pytorch
  ./make.sh

2 数据集

• 以VOC2012为例，将数据集建立软链接到data文件夹下：

  cd yolo2-pytorch
  mkdir data
  cd data
  ln -s "your VOCdevkit path" VOCdevkit2012

3 预训练权重

• 下载预训练权重darknet19
• 然后在yolo2-pytorch/cfgs/exps/darknet19_exp1.py中修改权重的路径。

训练

• 运行如下指令：

  python train.py

前向计算

• 在yolo2-pytorch/cfgs/config.py中修改trained_model的路径。

  mkdir output
  python test.py