fix: ruff lint errors

This commit is contained in:
blessedcoolant 2024-02-10 01:14:53 +05:30 committed by Kent Keirsey
parent 0a27b0379f
commit 675c73c94f
3 changed files with 5 additions and 5 deletions

View File

@ -61,7 +61,7 @@ class DWPoseDetector:
hands = candidate[:, 92:113]
hands = np.vstack([hands, candidate[:, 113:]])
bodies = dict(candidate=body, subset=score)
pose = dict(bodies=bodies, hands=hands, faces=faces)
bodies = {"candidate": body, "subset": score}
pose = {"bodies": bodies, "hands": hands, "faces": faces}
return draw_pose(pose, H, W, draw_face=draw_face, draw_hands=draw_hands, draw_body=draw_body)

View File

@ -62,7 +62,7 @@ def demo_postprocess(outputs, img_size, p6=False):
hsizes = [img_size[0] // stride for stride in strides]
wsizes = [img_size[1] // stride for stride in strides]
for hsize, wsize, stride in zip(hsizes, wsizes, strides):
for hsize, wsize, stride in zip(hsizes, wsizes, strides, strict=False):
xv, yv = np.meshgrid(np.arange(wsize), np.arange(hsize))
grid = np.stack((xv, yv), 2).reshape(1, -1, 2)
grids.append(grid)
@ -118,7 +118,7 @@ def inference_detector(session, oriImg):
final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]
isscore = final_scores > 0.3
iscat = final_cls_inds == 0
isbbox = [i and j for (i, j) in zip(isscore, iscat)]
isbbox = [i and j for (i, j) in zip(isscore, iscat, strict=False)]
final_boxes = final_boxes[isbbox]
else:
final_boxes = np.array([])

View File

@ -189,7 +189,7 @@ def draw_handpose(canvas, all_hand_peaks):
thickness=2,
)
for i, keyponit in enumerate(peaks):
for _, keyponit in enumerate(peaks):
x, y = keyponit
x = int(x * W)
y = int(y * H)