def parse_opt(): parser = argparse.ArgumentParser() parser.add_argument("--max_bz", type=int, default=10, help="batch_size of yolov5_model") parser.add_argument("--max_bz_pose", type=int, default=5, help="batch_size of pose_model") parser.add_argument("--max_num_processes", type=int, default=1) parser.add_argument("--logger_name_model", type=str, default="model") parser.add_argument("--logger_name_socket", type=str, default="socket") parser.add_argument("--device", type=int, default=0) parser.add_argument("--flip_test", action="store_true") parser.add_argument("--ip_port", type=int, default=8008) parser.add_argument("--mem_config", type=str, default="/home/cyliu35/teacher_behavior_callphone/app/src/mem_config.yml") parser.add_argument("--gap", type=int, default=5) parser.add_argument("--process_imgs_thres", type=int, default=21) parser.add_argument("--window_size", type=int, default=7, help="the size of the window to aggregate scores") parser.add_argument("--track", action="store_true", help="using tracking module") parser.add_argument("--visualization", action="store_true", help="whether to visualize the results") parser.add_argument("--block", action="store_true", help="whether to split the image to blocks") parser.add_argument( "--log_rotator", type=str, default="D", choices=["S", "M", "H", "D", "W0", "W1", "W2", "W3", "W4", "W5", "W6"]) parser.add_argument("--log_keep", type=int, default=7) return parser.parse_args()
时间: 2024-04-10 14:32:59 浏览: 158
socket.io-json-parser:基于JSON.stringify的socket.io解析器JSON.parse
这段代码是一个解析命令行参数的函数。它使用argparse模块创建一个ArgumentParser对象,并为该对象添加了一些参数。每个参数都有自己的类型、默认值和帮助信息。最后,函数使用parse_args()方法解析命令行参数,并返回一个包含参数值的命名空间对象。你可以根据需要修改参数的值和添加新的参数。
阅读全文