如何用 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是硬編碼值。