Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: ruff lint errors

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/image_util/dwpose/__init__.py
View File

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

4
invokeai/backend/image_util/dwpose/onnxdet.py
View File

@ -62,7 +62,7 @@ def demo_postprocess(outputs, img_size, p6=False):
hsizes = [img_size[0] // stride for stride in strides] hsizes = [img_size[0] // stride for stride in strides]
wsizes = [img_size[1] // 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)) xv, yv = np.meshgrid(np.arange(wsize), np.arange(hsize))
grid = np.stack((xv, yv), 2).reshape(1, -1, 2) grid = np.stack((xv, yv), 2).reshape(1, -1, 2)
grids.append(grid) 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] final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]
isscore = final_scores > 0.3 isscore = final_scores > 0.3
iscat = final_cls_inds == 0 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] final_boxes = final_boxes[isbbox]
else: else:
final_boxes = np.array([]) final_boxes = np.array([])

2
invokeai/backend/image_util/dwpose/utils.py
View File

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