如何用 python 中的输入替换命令行参数？

Question

我下载了代码并通过命令行参数运行它，但我想用原始输入修改它们，我使用的代码如下（要修改的部分代码）

if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--image_folder', required=True, help='path to image_folder which contains text images')
parser.add_argument('--workers', type=int, help='number of data loading workers', default=4)
parser.add_argument('--batch_size', type=int, default=192, help='input batch size')
parser.add_argument('--saved_model', required=True, help="path to saved_model to evaluation")
""" Data processing """
parser.add_argument('--batch_max_length', type=int, default=25, help='maximum-label-length')
parser.add_argument('--imgH', type=int, default=32, help='the height of the input image')
parser.add_argument('--imgW', type=int, default=100, help='the width of the input image')
parser.add_argument('--rgb', action='store_true', help='use rgb input')
parser.add_argument('--character', type=str, default='0123456789\"abcdefghijklmnopqrstuvwxyz\'ABCDEFGHIJKLMNOPQRS°TUV()WXY.Z', help='character label') #must be same as the characters in train.py
#'0123456789\"abcdefghijklmnopqrstuvwxyz\'ABCDEFGHIJKLMNOPQRS°TUV()WXY.Z'
parser.add_argument('--sensitive', action='store_true', help='for sensitive character mode')
parser.add_argument('--PAD', action='store_true', help='whether to keep ratio then pad for image resize')
""" Model Architecture """
parser.add_argument('--Transformation', type=str, required=True, help='Transformation stage. None|TPS')
parser.add_argument('--FeatureExtraction', type=str, required=True, help='FeatureExtraction stage. VGG|RCNN|ResNet')
parser.add_argument('--SequenceModeling', type=str, required=True, help='SequenceModeling stage. None|BiLSTM')
parser.add_argument('--Prediction', type=str, required=True, help='Prediction stage. CTC|Attn')
parser.add_argument('--num_fiducial', type=int, default=20, help='number of fiducial points of TPS-STN')
parser.add_argument('--input_channel', type=int, default=1, help='the number of input channel of Feature extractor')
parser.add_argument('--output_channel', type=int, default=512,
                    help='the number of output channel of Feature extractor')
parser.add_argument('--hidden_size', type=int, default=256, help='the size of the LSTM hidden state')

opt = parser.parse_args()

""" vocab / character number configuration """
if opt.sensitive:
    opt.character = string.printable[:-6]  # same with ASTER setting (use 94 char).

cudnn.benchmark = True
cudnn.deterministic = True
opt.num_gpu = torch.cuda.device_count()

demo(opt)

我使用的参数是python demo.py --image_folder eval --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --saved_model saved_models/TPS-ResNet-BiLSTM-Attn-Seed1111/best_accuracy.pth

现在我想用直接输入对其进行硬编码，我不知道该怎么做，请任何人帮助我。

提前致谢。

Answer 1

对于 argparse 中的每个参数，可以设置默认值，以便如果没有传递标志，它会回退到提供的值。

在您的代码中，您有： parser.add_argument('--output_channel', type=int, default=512)

注意default参数。 对于您想要硬编码的每个值，您可以将它们设置为默认值。

Answer 2

您可以指定默认值：

例如

parser.add_argument('--workers', type=int, help='number of data loading workers', default=4)

如果您不在命令行中修改--workers ，则default=4是硬编码值。