Merge pull request #1 from dwgeneral/dwgeneral-patch-1

Update jit_pose.py to fix out_bbox is NoneType issue
This commit is contained in:
Happy-Engineer
2024-07-10 14:54:58 +08:00
committed by GitHub
+2 -2
View File
@@ -22,7 +22,7 @@ def preprocess(
# get shape of image
img_shape = img.shape[:2]
out_img, out_center, out_scale = [], [], []
if len(out_bbox) == 0:
if out_bbox is None or len(out_bbox) == 0:
out_bbox = [[0, 0, img_shape[1], img_shape[0]]]
for i in range(len(out_bbox)):
x0 = out_bbox[i][0]
@@ -360,4 +360,4 @@ def inference_pose(model, out_bbox, oriImg, model_input_size=(288, 384)):
keypoints, scores = postprocess(outputs, model_input_size, center, scale)
return keypoints, scores
return keypoints, scores