若在bash的sh文件中,设置好cuda和多GPU
训练:
python trainval_net.py --dataset pascal_voc --net vgg16 --epochs 6 --bs 4 --nw 4
测试:
python test_net.py --dataset pascal_voc --net vgg16 --checkepoch 6 --checkpoint 3365
domo:
python demo.py --dataset pascal_voc --net vgg16 --checkepoch 6 --checkpoint 3365
若在命令行中:
训练:
python trainval_net.py --dataset pascal_voc --net vgg16 --epochs 6 --bs 4 --nw 4 --cuda --mGPUs
测试:
python test_net.py --dataset pascal_voc --net vgg16 --checkepoch 6 --checkpoint 3365 --cuda --mGPUs
domo:
python demo.py --dataset pascal_voc --net vgg16 --checkepoch 6 --checkpoint 3365 --cuda --mGPUs