Skip to content

Commit 2f4bbf7

Browse files
committed
Fix lint
1 parent ea3f722 commit 2f4bbf7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/pytorch/object-detector/predictor.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,9 @@ def predict(self, payload):
3232
pred = self.model(img_tensor)
3333

3434
predicted_class = [self.coco_labels[i] for i in list(pred[0]["labels"].numpy())]
35-
predicted_boxes = [[(i[0], i[1]), (i[2], i[3])] for i in list(pred[0]["boxes"].detach().numpy())]
35+
predicted_boxes = [
36+
[(i[0], i[1]), (i[2], i[3])] for i in list(pred[0]["boxes"].detach().numpy())
37+
]
3638
predicted_score = list(pred[0]["scores"].detach().numpy())
3739
predicted_t = [predicted_score.index(x) for x in predicted_score if x > threshold]
3840
if len(predicted_t) == 0:

0 commit comments

Comments
 (0)