|
|
@@ -15,7 +15,7 @@ parser.add_argument('--normalization', default='sigmoid', type=str, help='|选
|
|
|
parser.add_argument('--batch', default=1, type=int, help='|输入图片批量|')
|
|
|
parser.add_argument('--device', default='cuda', type=str, help='|推理设备|')
|
|
|
parser.add_argument('--num_worker', default=0, type=int, help='|CPU处理数据的进程数，0只有一个主进程，一般为0、2、4、8|')
|
|
|
-parser.add_argument('--float16', default=True, type=bool, help='|推理数据类型，要支持float16的GPU，False时为float32|')
|
|
|
+parser.add_argument('--float16', default=False, type=bool, help='|推理数据类型，要支持float16的GPU，False时为float32|')
|
|
|
args, _ = parser.parse_known_args() # 防止传入参数冲突，替代args = parser.parse_args()
|
|
|
# -------------------------------------------------------------------------------------------------------------------- #
|
|
|
assert os.path.exists(args.model_path), f'! model_path不存在:{args.model_path} !'
|
