Python-如何在IDE中使用不同的命令行參數調用腳本？

Question

我正在為我的最后一年的項目使用神經網絡進行裸露檢測。 到目前為止，我找到了一個github鏈接來完成我的項目。 要運行此項目，我需要傳遞2個參數，例如input_file和-m/--model_weights 。 我正在使用Pycharm IDE。 我發現了一個SO回答傳遞參數。 但就我而言，我需要使用兩行代碼來傳遞參數。 不像在pycharm的“腳本參數”框中提供參數。 我該怎么做？

input_file表示圖像的路徑。

-m/--model_weights表示權重的路徑。 在我的情況下，權重路徑為data/nsw_weights.npy.

這是我的項目文件。

import sys
import argparse
import tensorflow as tf

from model import OpenNsfwModel, InputType
from image_utils import create_tensorflow_image_loader
from image_utils import create_yahoo_image_loader

import numpy as np


IMAGE_LOADER_TENSORFLOW = "tensorflow"
IMAGE_LOADER_YAHOO = "yahoo"


def main(argv):
    parser = argparse.ArgumentParser()

    parser.add_argument("input_file", help="Path to the input image.\
                        Only jpeg images are supported.")
    parser.add_argument("-m", "--model_weights", required=True,
                        help="Path to trained model weights file")

    parser.add_argument("-l", "--image_loader",
                        default=IMAGE_LOADER_YAHOO,
                        help="image loading mechanism",
                        choices=[IMAGE_LOADER_YAHOO, IMAGE_LOADER_TENSORFLOW])

    parser.add_argument("-t", "--input_type",
                        default=InputType.TENSOR.name.lower(),
                        help="input type",
                        choices=[InputType.TENSOR.name.lower(),
                                 InputType.BASE64_JPEG.name.lower()])

    args = parser.parse_args()

    model = OpenNsfwModel()

    with tf.Session() as sess:

        input_type = InputType[args.input_type.upper()]
        model.build(weights_path=args.model_weights, input_type=input_type)

        fn_load_image = None

        if input_type == InputType.TENSOR:
            if args.image_loader == IMAGE_LOADER_TENSORFLOW:
                fn_load_image = create_tensorflow_image_loader(sess)
            else:
                fn_load_image = create_yahoo_image_loader()
        elif input_type == InputType.BASE64_JPEG:
            import base64
            fn_load_image = lambda filename: np.array([base64.urlsafe_b64encode(open(filename, "rb").read())])

        sess.run(tf.global_variables_initializer())

        image = fn_load_image(args.input_file)

        predictions = \
            sess.run(model.predictions,
                     feed_dict={model.input: image})

        print("Results for '{}'".format(args.input_file))
        print("\tSFW score:\t{}\n\tNSFW score:\t{}".format(*predictions[0]))

if __name__ == "__main__":
    main(sys.argv)

Answer 1

在pycharm中，您可以設置多個運行配置，以允許您傳遞不同的參數。

該文檔的一節介紹了設置方法。

如果您有一組要確認結果的特定輸入，則可能還需要考慮編寫測試。