假设要运行'PycharmProjects/faster-rcnn/faster-rcnn.pytorch-pytorch-1.0'路径下的demo.py文件,则存在两种可行方式。
--net vgg16:backbone网络选择vgg16
--cuda:启用gpu训练
--load_dir models:在models文件夹下加载所需的文件
在方式一的第四步中,如果需要赋值的参数很多,我们每次运行demo.py文件时都要重新赋值,未免有些繁琐。为了解决这个问题,我们可以在demo.py同目录下新建一个demo.sh文件,并输入以下内容:
python demo.py --net vgg16 \
--cuda \
--load_dir models
此时,我们只需要在方式一的第四步中,运行demo.sh文件即可: