使用SSD pytorch进行VOC数据集的训练

SSD-pytorch的训练

使用git

git clone https://github.com/lufficc/SSD

配置环境时，需进行以下命令：

pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu80/stable

其中80根据自己的cuda包改对应的数字，如果是cuda9.2,就改成92.
我使用的是pytorch1.1.0,所以如下输入：

pip install torch==1.1.0 -f https://download.pytorch.org/whl/cu92/stable

如果使用torchvision 0.3以上，就不需要NMS否则

cd ext
	
# build nms
cd ext
python build.py build_ext develop

注意datasets应如下命名：

~/datasets/VOC2007...

单GPU输入如下命令：

python train.py --config-file configs/vgg_ssd300_voc0712.yaml

多GPU：

export NGPUS=4
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py --config-file configs/vgg_ssd300_voc0712.yaml SOLVER.WARMUP_FACTOR 0.03333 SOLVER.WARMUP_ITERS 1000

当然你可以修改数据集，目录在~configs/，打开后，随便打开一个如下：

我用不到VOC2012，所以删掉了。但记住保留后面的","，否则会报错。

输入命令回车，训练就成功了

已经开始迭代