diff --git a/deploy/python/det_keypoint_unite_utils.py b/deploy/python/det_keypoint_unite_utils.py index ccd4a9b45bf57ec5961e4ecaa28504d12a815a67..26344628a3e10457a394f351fc64f7049a4245bb 100644 --- a/deploy/python/det_keypoint_unite_utils.py +++ b/deploy/python/det_keypoint_unite_utils.py @@ -112,7 +112,7 @@ def argsparser(): "calibration, trt_calib_mode need to set True.") parser.add_argument( '--use_dark', - type=bool, + type=ast.literal_eval, default=True, help='whether to use darkpose to get better keypoint position predict ') parser.add_argument( diff --git a/deploy/python/utils.py b/deploy/python/utils.py index 1ab86b1f4e0067e6006baec0f96fc42d092f5ad2..a0f8e352dd6af22696bf9f157b76aff66199b311 100644 --- a/deploy/python/utils.py +++ b/deploy/python/utils.py @@ -133,7 +133,7 @@ def argsparser(): help="max batch_size for reid model inference.") parser.add_argument( '--use_dark', - type=bool, + type=ast.literal_eval, default=True, help='whether to use darkpose to get better keypoint position predict ') return parser