使用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/,打开后,随便打开一个如下:
使用SSD pytorch进行VOC数据集的训练_第1张图片
我用不到VOC2012,所以删掉了。但记住保留后面的",",否则会报错。

输入命令回车,训练就成功了
在这里插入图片描述

已经开始迭代

你可能感兴趣的:(使用SSD pytorch进行VOC数据集的训练)