直接使用train-.py训练命令: python tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/faster_rcnn_models/ZF_faster_rcnn_final.caffemodel --cfg experiments/cfgs/faster_rcnn_alt_opt.yml --imdb voc_2007_trainval
使用sh脚本程序来训练,更简单且能保存日志,输入参数在该sh文件下修改
四步训练 alt-train: bash ./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc
不使用脚本:sudo python ./tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/imagenet_models/ZF_faster_rcnn_final.caffemodel --imdb voc_2007_trainval --cfg experiments/cfgs/faster_rcnn_alt_opt.yml
端对端训练 end2end-train: sudo bash ./experiments/scripts/faster_rcnn_end2end.sh 0 ZF pascal_voc
测试命令: python tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt --net output/faster_rcnn_alt_opt/voc_2007_trainval/ZF_faster_rcnn_final.caffemodel --cfg experiments/cfgs/faster_rcnn_alt_opt.yml --imdb voc_2007_test
python tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_end2end/test.prototxt --net output/faster_rcnn_end2end/voc_2007_trainval/zf_faster_rcnn_iter_50000.caffemodel --cfg experiments/cfgs/faster_rcnn_end2end.yml --imdb voc_2007_test
测试命令VGG:python tools/test_net.py --gpu 0 --def models/pascal_voc/VGG16/faster_rcnn_end2end/test.prototxt --net output/faster_rcnn_end2end/voc_2007_trainval/vgg16_faster_rcnn_iter_40000.caffemodel --cfg experiments/cfgs/faster_rcnn_end2end.yml --imdb voc_2007_test
检测命令: ./tools/demo.py --gpu 0 --net zf
特征可视化命令:python tools/feature_visualize.py --net zf (--net vgg16)
添加执行权限 sudo chmod 777 py-faster-rcnn/*
sudo chmod -R 777 py-faster-rcnn
特征可视化:
python tools/feature_visualize.py
监视显存
我们设置为每 10s 显示一次显存的情况
watch -n 2 nvidia-smi
使用sh脚本程序来训练,更简单且能保存日志,输入参数在该sh文件下修改
四步训练 alt-train: bash ./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc
不使用脚本:sudo python ./tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/imagenet_models/ZF_faster_rcnn_final.caffemodel --imdb voc_2007_trainval --cfg experiments/cfgs/faster_rcnn_alt_opt.yml
端对端训练 end2end-train: sudo bash ./experiments/scripts/faster_rcnn_end2end.sh 0 ZF pascal_voc
测试命令: python tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt --net output/faster_rcnn_alt_opt/voc_2007_trainval/ZF_faster_rcnn_final.caffemodel --cfg experiments/cfgs/faster_rcnn_alt_opt.yml --imdb voc_2007_test
python tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_end2end/test.prototxt --net output/faster_rcnn_end2end/voc_2007_trainval/zf_faster_rcnn_iter_50000.caffemodel --cfg experiments/cfgs/faster_rcnn_end2end.yml --imdb voc_2007_test
测试命令VGG:python tools/test_net.py --gpu 0 --def models/pascal_voc/VGG16/faster_rcnn_end2end/test.prototxt --net output/faster_rcnn_end2end/voc_2007_trainval/vgg16_faster_rcnn_iter_40000.caffemodel --cfg experiments/cfgs/faster_rcnn_end2end.yml --imdb voc_2007_test
检测命令: ./tools/demo.py --gpu 0 --net zf
特征可视化命令:python tools/feature_visualize.py --net zf (--net vgg16)
添加执行权限 sudo chmod 777 py-faster-rcnn/*
sudo chmod -R 777 py-faster-rcnn
特征可视化:
python tools/feature_visualize.py
监视显存
我们设置为每 10s 显示一次显存的情况
watch -n 2 nvidia-smi