深度学习的工程中,py文件中的main函数一开始总会有大量的参数传入,而通常使用的方法就是使用argparse通过命令行传入,这篇博文旨在记录argparse的常用方法。
一、语句简介
1.载入argparse模块:
import argparse
2.创建语法分析器:
parser = argparse.ArgumentParser()
3.添加选项:
parser.add_argument("--epochs", type=int, default=100, help="number of epochs")
##
parser.add_argument("--", type=, default=, help="")
4.实例化语法分析器:
opt = parser.parse_args()
5.调用:
opt.epochs
二、标准使用方式
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser()
# Super-Params
parser.add_argument("--epochs", type=int, default=100, help="number of epochs")
parser.add_argument("--batch_size", type=int, default=64, help="size of each image batch")
parser.add_argument("--img_size", type=int, default=416, help="size of each image dimension")
parser.add_argument("--n_cpu", type=int, default=8, help="number of cpu threads to use during batch generation")
parser.add_argument("--gradient_accumulations", type=int, default=2, help="number of gradient accums before step")
# Model config
parser.add_argument("--model_def", type=str, default="config/yolov3-tiny_lsq.cfg", help="path to definition file")
parser.add_argument("--pretrained_weights", type=str, help="if specified starts from checkpoint model")
parser.add_argument("--gpu", type=int, default=1, help="assign a gpu to this porject, start from 0")
# Eval & Save interval
parser.add_argument("--checkpoint_interval", type=int, default=1, help="interval between saving model weights")
parser.add_argument("--evaluation_interval", type=int, default=2, help="interval evaluations on validation set")
# Dataset info
parser.add_argument("--data_config", type=str, default="config/dac.data", help="path to data config file")
# Others
parser.add_argument("--compute_map", default=False, help="if True computes mAP every tenth batch")
parser.add_argument("--multiscale_training", default=True, help="allow for multi-scale training")
# Quantisize Settings:
parser.add_argument("--quan_weitht_bit", type=int, default=8, help="")
parser.add_argument("--quan_activation_bit", type=int, default=8, help="")
opt = parser.parse_args()
print(opt)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)