From 675c73c94fd6a859146de7e762e2ec30eed9ad30 Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Sat, 10 Feb 2024 01:14:53 +0530 Subject: [PATCH] fix: ruff lint errors --- invokeai/backend/image_util/dwpose/__init__.py | 4 ++-- invokeai/backend/image_util/dwpose/onnxdet.py | 4 ++-- invokeai/backend/image_util/dwpose/utils.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/invokeai/backend/image_util/dwpose/__init__.py b/invokeai/backend/image_util/dwpose/__init__.py index 1ae8c6cd0d..05c0687247 100644 --- a/invokeai/backend/image_util/dwpose/__init__.py +++ b/invokeai/backend/image_util/dwpose/__init__.py @@ -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) diff --git a/invokeai/backend/image_util/dwpose/onnxdet.py b/invokeai/backend/image_util/dwpose/onnxdet.py index cf2706b596..44aca3aac8 100644 --- a/invokeai/backend/image_util/dwpose/onnxdet.py +++ b/invokeai/backend/image_util/dwpose/onnxdet.py @@ -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([]) diff --git a/invokeai/backend/image_util/dwpose/utils.py b/invokeai/backend/image_util/dwpose/utils.py index aa98efdb29..409e4a9837 100644 --- a/invokeai/backend/image_util/dwpose/utils.py +++ b/invokeai/backend/image_util/dwpose/utils.py @@ -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)