赞
踩
Yolov-1-TX2上用YOLOv3训练自己数据集的流程(VOC2007-TX2-GPU)
Yolov--2--一文全面了解深度学习性能优化加速引擎---TensorRT
Yolov--3--TensorRT中yolov3性能优化加速(基于caffe)
yolov--10--目标检测模型的参数评估指标详解、概念解析
yolov--11--YOLO v3的原版训练记录、mAP、AP、recall、precision、time等评价指标计算
yolov--12--YOLOv3的原理深度剖析和关键点讲解
16、开发环境个人单独配置
source activate pytorch1.1-cuda9-py3.6
source activate pytorch1.1.0-py2.7_cuda9.0
https://github.com/Stinky-Tofu/Stronger-yolo/tree/master/v3
CUDA_VISIBLE_DEVICES='4' python train.py &
14、Detectron-e2e_mask_rcnn_R-101-FPN_2x_gn
训练:
CUDA_VISIBLE_DEVICES='8,9' python tools/train_net.py --cfg experiments/e2e_mask_rcnn_R-101-FPN_1x-train1999-2gpu.yaml OUTPUT_DIR e2e_mask_rcnn_R-101-FPN_1x-train1999-2gpu/ | tee visualization/e2e_mask_rcnn_R-101-FPN_1x-train1999-2gpu-start8-21-1620-833-500-1.log
测试:
训练:
测试:
训练:
CUDA_VISIBLE_DEVICES='6,7' python tools/train_net.py --cfg experiments/e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu.yaml OUTPUT_DIR e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu/ | tee visualization/e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu-start8-21-1520-833-500-1.log
测试:
CUDA_VISIBLE_DEVICES=6 python tools/test_net.py --cfg experiments/e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu.yaml TEST.WEIGHTS e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu/train/voc_1999_train\:voc_1999_val/generalized_rcnn/model_final.pkl NUM_GPUS 1 | tee visualization/e2e_faster_rcnn_R-101-FPN_1x-train1999-2gpu-test_net-1.log
训练:
CUDA_VISIBLE_DEVICES='6,7' python tools/train_net.py --cfg experiments/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN-train1999.yaml OUTPUT_DIR tutorial_2gpu_e2e_faster_rcnn_R-50-FPN-train1999/ | tee visualization/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN-train1999-start8-21-1355-2gpu-833-500-1.log
测试:
https://blog.csdn.net/chengyq116/article/details/83213699
注意:测试计算map时区分val.txt与test.txt ,差别较大,在2-4%
训练:
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg yolov3-tiny.conv.22 -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-2/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-start2019-8-22-2050-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg yolov3-tiny.conv.22 -gpus 6,7 | tee visualization-tiny-v3-a-voc1999-iter8000-mixup=1/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-start2019-8-24-0110-mixup-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg yolov3-tiny.conv.22 -gpus 6,7 | tee visualization-tiny-v3-a-voc1999-iter8000-angle=1/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-start2019-8-24-0140-angle-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999-iter8000--mixup-angle=1/yolov3-tiny-voc1999_last.weights -gpus 6,7 | tee visualization-tiny-v3-a-voc1999-iter12000--mixup-angle=1/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera12000-start2019-8-24-1810-angle-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup-flip.cfg yolov3-tiny.conv.22 -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-iter8000-mixup-flip-1/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-mixup-flip-start2019-8-26-1530-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup-flip.cfg backup-tiny-v3-a-voc1999-iter8000-mixup-flip-1/yolov3-tiny-voc1999-mixup-flip_last.weights -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-iter8000-mixup-flip-1/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-mixup-flip-start2019-8-26-1530-2.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup-flip.cfg yolov3-tiny.conv.22 -dont_show -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-iter10000-mixup-flip-1/tiny-v3-a-voc1999-bach64-4-2lei-train723-val181-itera10000-mixup-flip-start2019-8-27-1050-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup-flip.cfg backup-tiny-v3-a-voc1999-iter10000-mixup-flip-1/yolov3-tiny-voc1999-mixup-flip_last.weights -dont_show -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-iter10000-mixup-flip-1/tiny-v3-a-voc1999-bach64-4-2lei-train723-val181-itera10000-mixup-flip-start2019-8-27-1050-2.log
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-4.cfg backup-tiny-v3-a-voc1999-4/yolov3-tiny-voc1999-4_last.weights -dont_show -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-4/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera5000-start2019-12-3-1810-2.log
tiny-v3-a-voc1999-mixup
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup.cfg backup-tiny-v3-a-voc1999-iter8000-mixup=1/yolov3-tiny-voc1999_final.weights -points 180 | tee visualization-tiny-v3-a-voc1999-iter8000-mixup=1/tiny-v3-voc1999-bach64-16-2lei-test181-prcurves-mixup-180points-iter8000-finalweights.log
tiny-v3-a-voc1999-3s
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-3l.cfg yolov3-tiny.conv.22 -dont_show -gpus 8,9 | tee visualization-tiny-v3-a-voc1999-iter8000-3l/tiny-v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-31-start2019-11-23-1930.log
tiny-v3-a-voc1999-3s-mixup
./darknet detector train cfg/voc1999.data cfg/yolov3-tiny-voc1999-3l-mixup.cfg yolov3-tiny.conv.22 -dont_show -gpus 6,7 | tee visualization-tiny-v3-a-voc1999-iter8000-3l-mixup/tiny-v3-a-voc1999-bach64-4-2lei-train723-val181-itera8000-31-mixup-start2019-11-23-2050.log
计算map:
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999/yolov3-tiny-voc1999_final.weights -points 11 | tee visualization-tiny-v3-a-voc1999/tiny-v3-voc1999-bach64-4-2lei-test181-prcurves-11points-finalweights.log
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999/yolov3-tiny-voc1999_9000.weights -points 0 | tee visualization-tiny-v3-a-voc1999/tiny-v3-voc1999-bach64-4-2lei-test181-prcurves-0points-9000weights.log
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999-iter8000-angle=1/yolov3-tiny-voc1999_final.weights -points 101 | tee visualization-tiny-v3-a-voc1999-iter8000-angle=1/tiny-v3-voc1999-bach64-16-2lei-test181-prcurves-101points-iter8000-angle-finalweights.log
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999-iter12000--mixup-angle=1/yolov3-tiny-voc1999_final.weights -points 50 | tee visualization-tiny-v3-a-voc1999-iter12000--mixup-angle=1/tiny-v3-voc1999-bach64-16-2lei-test181-prcurves-50points-iter12000-mixup-angle-finalweights.log
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999-mixup-flip.cfg backup-tiny-v3-a-voc1999-iter8000-mixup-flip-1/yolov3-tiny-voc1999-mixup-flip_final.weights -points 50 | tee visualization-tiny-v3-a-voc1999-iter8000-mixup-flip-1/tiny-v3-voc1999-bach64-16-2lei-test181-prcurves-50points-iter8000-mixup-flip-finalweights.log
tiny-v3-a-voc1999-3s
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999-3l.cfg backup-tiny-v3-a-voc1999-iter8000-3l/yolov3-tiny-voc1999-3l_final.weights -points 0 | tee visualization-tiny-v3-a-voc1999-iter8000-3l/tiny-v3-voc1999-bach64-16-2lei-test181-prcurves-0points-iter8000-3l-finalweights.log
tiny-v3-a-voc1999-3s-mixup
./darknet detector map cfg/voc1999.data cfg/yolov3-tiny-voc1999-3l-mixup.cfg backup-tiny-v3-a-voc1999-iter8000-3l-mixup/yolov3-tiny-voc1999-3l-mixup_final.weights -points 0 | tee visualization-tiny-v3-a-voc1999-iter8000-3l-mixup/tiny-v3-voc1999-bach64-4-2lei-test181-prcurves-0points-iter8000-3l-mixup-finalweights.log
测试单张:
./darknet detector test cfg/voc1999.data cfg/yolov3-tiny-voc1999.cfg backup-tiny-v3-a-voc1999-2/yolov3-tiny-voc1999_final.weights scripts/VOCdevkit/VOC1999/JPEGImages/000001.jpg | tee visualization-tiny-v3-test/test-181-tiny-v3-voc1999-bach64-16-2lei-iter8000-000001-jpg-1.log
批量测试:
训练:
./darknet detector train cfg/voc1999.data cfg/yolov3-a-voc1999.cfg darknet53.conv.74 -gpus 6,7 | tee visualization-1999/v3-a-voc1999-bach64-16-2lei-train723-val181-itera-start2019-8-19-1050-1.log
./darknet detector train cfg/voc1999.data cfg/yolov3-a-voc1999.cfg darknet53.conv.74 -gpus 6,7 | tee visualization-1999-2/v3-a-voc1999-bach64-16-2lei-train723-val181-itera8000-start2019-8-21-2250-1.log
计算map:
./darknet detector map cfg/voc1999.data cfg/yolov3-a-voc1999.cfg backup-voc1999/yolov3-a-voc1999_final.weights -points 0 | tee visualization-1999/v3-a-voc1999-bach64-16-2lei-test181-prcurves-0points-finalweights.log
./darknet detector map cfg/voc1999.data cfg/yolov3-a-voc1999.cfg backup-voc1999-2/yolov3-a-voc1999_final.weights -points 0 | tee visualization-1999-2-ite8000/v3-a-voc1999-bach64-16-2lei-test181-prcurves-0points-iter8000-finalweights.log
测试:
./darknet detector test cfg/voc1999.data cfg/yolov3-a-voc1999.cfg backup-voc1999-2/yolov3-a-voc1999_final.weights scripts/VOCdevkit/VOC1999/JPEGImages/000001.jpg | tee visualization-1999-ite8000-test/test-181-v3-voc1999-bach64-16-2lei-iter8000-000001-jpg-1.log
批量测试:
./darknet detector test cfg/voc1999.data cfg/yolov3-a-voc1999.cfg backup-voc1999-2/yolov3-a-voc1999_final.weights -dont_show -ext_output < scripts/1999_test.txt > results/1999-ite8000-test.txt | tee visualization-1999-ite8000-test/test-181-v3-voc1999-bach64-16-2lei-iter8000-jpg-1.log
1.对原始weights文件进行稀疏化训练
python sparsity_train-3.py -gpus 6,7 | tee visualization/sparsity-07-12trainval-yolov3-a-0.0001-resize416-batch128-32-2gpu-start2019-7-28-1450-1.log
CUDA_VISIBLE_DEVICES='7' python sparsity_train-3.py | tee visualization/sparsity-07-12trainval-yolov3-a-0.0001-resize416-batch16-16-3cpu-start2019-7-28-1520-1.log
CUDA_VISIBLE_DEVICES='7' python sparsity_train-3.py -sr --s 0.001 | tee visualization/sparsity-07trainval-yolov3-a-2-0.001-resize416-batch16-1-2cpu-start2019-8-3-1520-1.log
2.剪枝
CUDA_VISIBLE_DEVICES=9 python new_prune-0.3.py --weights checkpoints-q-1/yolov3_sparsity_416_0.0001_final_1_182.weights --percent 0.5 | tee visualization/prune-0.5-07-12trainval-yolov3-a-resize416-batch16-1-2cpu-start2019-8-4-1500-1.log
3.对剪枝后的weights进行微调
CUDA_VISIBLE_DEVICES='7' python sparsity_train-weitiao.py --cfg prune_yolov3-voc-sparsity-1.cfg --weights checkpoints-q-1/prune_yolov3_sparsity_416_0.0001_final_1_182.weights | tee visualization/weitiao-07-12trainval-yolov3-a-resize416-batch16-1-2cpu-start2019-8-4-1500-1.log
4.微调后进行测试:
4.微调后进行测试:
https://github.com/facebookresearch/Detectronhttps://caffe2.ai/docs/tutorials.html
https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md
CUDA_VISIBLE_DEVICES='3,4' python tools/train_net.py --cfg experiment/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007.yaml OUTPUT_DIR out-1 | tee visualization/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007.log
CUDA_VISIBLE_DEVICES=2 python tools/train_net.py --cfg experiments/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007.yaml OUTPUT_DIR out-faster-rcnn-1/train-2 | tee visualization/1gpu_e2e_faster_rcnn_R-50-FPN-voc2007-iter70000-start-7-20-2150.log
2019-7-25:
CUDA_VISIBLE_DEVICES='2,3' python tools/train_net.py --cfg experiments/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN-trainval07+12.yaml OUTPUT_DIR tutorial_2gpu_e2e_faster_rcnn_R-50-FPN/ | tee visualization/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN-trainval07+12-ite60000-1.log
CUDA_VISIBLE_DEVICES='6,7,8' python tools/train_net.py --cfg experiments/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12.yaml OUTPUT_DIR e2e_faster_rcnn_R-101-FPN_1x-trainval07+12/train-3/ | tee visualization/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-ite90000-start7-26-2040-1.log
CUDA_VISIBLE_DEVICES='6,7' python tools/train_net.py --cfg experiments/e2e_mask_rcnn_R-50-FPN_2x_gn-trainval07+12.yaml OUTPUT_DIR e2e_mask_rcnn_R-50-FPN_2x_gn-trainval07+12/ | tee visualization/e2e_mask_rcnn_R-50-FPN_2x_gn-trainval07+12-start7-27-1200-2gpu-1.log
CUDA_VISIBLE_DEVICES='8,9' python tools/train_net.py --cfg experiments/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-2gpu.yaml OUTPUT_DIR e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-2gpu/ | tee visualization/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-2gpu-start7-27-1215-1.log
CUDA_VISIBLE_DEVICES='6,7,8,9' python tools/train_net.py --cfg experiments/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-4gpu.yaml OUTPUT_DIR e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-4gpu-iter20000/ | tee visualization/e2e_faster_rcnn_R-101-FPN_1x-trainval07+12-4gpu-start7-28-1100-iter20000-1.log
CUDA_VISIBLE_DEVICES=4 python tools/infer_simple.py --cfg experiment/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007.yaml --output-dir out-1/detectron-visualizations/ --image-ext jpg --wts out-1/train/voc_2007_train/generalized_rcnn/model_final.pkl demo | tee visualization/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007-infer-simple.log
CUDA_VISIBLE_DEVICES=4 python tools/test_net.py --cfg experiments/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007.yaml TEST.WEIGHTS out-faster-rcnn-1/train/voc_2007_train/generalized_rcnn/model_final.pkl NUM_GPUS 1 | tee visualization/2gpu_e2e_faster_rcnn_R-50-FPN-voc2007-test_net.log
CUDA_VISIBLE_DEVICES=4 python tools/visualize_results.py --dataset voc_2007_val --detections out-faster-rcnn-1/test/voc_2007_val/generalized_rcnn/detections.pkl --output-dir out-faster-rcnn-1/detectron-visualizations
问题:
FileNotFoundError: [Errno 2] No such file or directory: '/home/xxx/detectron/datasets/data/VOC2007/VOCdevkit2007/results/VOC2007/Main/comp4_80ce9d5d-3ac9-4469-8914-2b6eabc81797_det_val_aeroplane.txt'
新建results/VOC2007/Main文件夹即可
训练:
CUDA_VISIBLE_DEVICES='2,3' python tools/train_net.py --cfg experiments/e2e_mask_rcnn_R-50-FPN_2x_gn.yaml OUTPUT_DIR out-e2e_mask_rcnn_R-50-FPN_2x_gn/ | tee visualization/e2e_mask_rcnn_R-50-FPN_2x_gn-voc2007-iter180000-start-7-21-1135.log
CUDA_VISIBLE_DEVICES='4,5' python tools/train_net.py --cfg experiments/e2e_mask_rcnn_R-101-FPN_1x.yaml OUTPUT_DIR out-e2e_mask_rcnn_R-101-FPN_1x/ | tee visualization/e2e_mask_rcnn_R-101-FPN_1x-voc2007-iter90000-start-7-21-1935.log
CUDA_VISIBLE_DEVICES=8 ./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg darknet53.conv.74 -map
原版训练:
CUDA_VISIBLE_DEVICES=3 ./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg backup-y/yolov3-voc-y_2500.weights | tee visualization/train-yolov3-1.log
CUDA_VISIBLE_DEVICES=4 ./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg | tee visualization/train-yolov3-1.log
CUDA_VISIBLE_DEVICES=4 ./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg backup/yolov3-voc-y_last.weights 2>1 | tee visualization/darknet-y-batch64-16-voc07-train5011-iteration16767-02.log
voc2007+12trainval训练:
./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg darknet53.conv.74 -gpus 4,5 | tee visualization/darknet-y-batch128-32-voc0712trainval-iteration-start7-24-1300-01.log
./darknet detector train cfg/voc-y.data cfg/yolov3-voc-y.cfg backup/yolov3-voc-y_last.weights -gpus 4,5 | tee visualization/darknet-y-batch128-32-voc0712trainval-iteration-start7-24-1300-02.log
计算map:
./darknet detector map cfg/voc-y.data cfg/yolov3-voc-y.cfg backup/yolov3-voc-y_41000.weights | tee visualization/darknet-voc2007-bach64-16-20lei-iteration41000-prcurves-points-41000weights.log
CUDA_VISIBLE_DEVICES='7' ./darknet detector map cfg/voc-y.data cfg/yolov3-voc-y.cfg backup/yolov3-voc-y_final.weights | tee visualization/darknet-a-y-batch128-32-voc0712trainval-iteration60200-test-1.log
原版检测单张图片:
CUDA_VISIBLE_DEVICES=8 ./darknet detector test ./cfg/voc-y.data ./cfg/yolov3-voc-y.cfg ./yolov3.weights data/dog.jpg
训练:
CUDA_VISIBLE_DEVICES=4 ./darknet detector train cfg/voc2001.data cfg/yolov3-voc2001.cfg darknet53.conv.74 | tee visualization/darknet-voc2001-bach64-32-2lei-train5011-iteration.log
计算map:
./darknet detector map cfg/voc2001.data cfg/yolov3-voc2001.cfg backup/yolov3-voc2001_5000.weights | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration-5000weights.log
./darknet detector map cfg/voc2001.data cfg/yolov3-voc2001.cfg backup/yolov3-voc2001_4000.weights | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration5000-4000weights.log
./darknet detector map cfg/voc2001.data cfg/yolov3-voc2001.cfg backup/yolov3-voc2001_3000.weights | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration5000-3000weights.log
./darknet detector map cfg/voc2001.data cfg/yolov3-voc2001.cfg backup/yolov3-voc2001_5000.weights -points 11 | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration-prcurves-11points-5000weights.log
检测单张图片:
./darknet detector test cfg/voc2001.data cfg/yolov3-voc2001.cfg backup/yolov3-voc2001_last.weights data/000001.jpg | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration5000-000001-jpg.log
训练:
CUDA_VISIBLE_DEVICES=5 ./darknet detector train cfg/voc2001.data cfg/yolov2-voc2001.cfg darknet19_448.conv.23 | tee visualization/darknet-v2-voc2001-batch64-16-trainval904-iteration-start6-30-22:58-01.log
计算map:
./darknet detector map cfg/voc2001.data cfg/yolov2-voc2001.cfg backup/yolov2-voc2001_4000.weights | tee visualization/darknet-v2-voc2001-bach64-32-2lei-trainval904-iteration5000-4000weights.log
./darknet detector map cfg/voc2001.data cfg/yolov2-voc2001.cfg backup/yolov2-voc2001_final.weights | tee visualization/darknet-v2-voc2001-bach64-32-2lei-trainval904-iteration5000-finalweights.log
检测单张图片:
./darknet detector test cfg/voc2001.data cfg/yolov2-voc2001.cfg backup/yolov2-voc2001_last.weights data/000002.jpg | tee visualization/darknet-voc2001-bach64-32-2lei-trainval904-iteration5000-000002-jpg.log
CUDA_VISIBLE_DEVICES=8 python train.py | tee visualization/train-ssd300-voc07-batch32-classes21-iter-04.log
稀疏化:
CUDA_VISIBLE_DEVICES=9 python sparsity_train-3.py | tee visualization/sparsity-train-yolov3-voc07trainval--02.log
剪枝:
CUDA_VISIBLE_DEVICES=9 python new_prune-0.3.py --weights checkpoints-3/yolov3_sparsity_416_0.0001_final_1_410.weights
训练:
CUDA_VISIBLE_DEVICES=4 python trainval_net.py --bs 10 --cuda --checkepoch 13 --checkpoint 1001 --use_tfb | tee visualization/faster-rcnn-voc07-trainval-iteration-01.log
测试:
CUDA_VISIBLE_DEVICES=4 python test_net.py --checkepoch 20 --checkpoint 1001 --cuda | tee visualization/faster-rcnn-voc07-trainval-checkepoch-20-checkpoint-1001-01.log
2、自己数据集
训练:
python setup.py build develop
CUDA_VISIBLE_DEVICES=4 python trainval_net.py --bs 5 --cuda --use_tfb | tee visualization/faster-rcnn-voc2001-trainval-bs5-iteration-01.log
测试:
CUDA_VISIBLE_DEVICES=4 python test_net.py --checkepoch 20 --checkpoint 179 --cuda | tee visualization/faster-rcnn-voc2001-test275-checkepoch-20-checkpoint-179-01.log
1、Getting Started
anaconda创建pytorch1.1虚拟环境:
source activate pytorch
退出pytorch1.1虚拟环境:
conda deactivate
conda create --name cornerNet_Lite-y --file conda_packagelist.txt --channel pytorch
https://conda.anaconda.org/pytorch/linux-64/pytorch-1.0.0-py3.7_cuda10.0.130_cudnn7.4.1_1.tar.bz2
原文:https://blog.csdn.net/u014236392/article/details/81127537
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。