Spaces:
Sleeping
Sleeping
nicer
Browse files
maskrcnn_benchmark/engine/predictor_glip.py
CHANGED
@@ -269,7 +269,7 @@ class GLIPDemo(object):
|
|
269 |
for box, color in zip(boxes, colors):
|
270 |
box = box.to(torch.int64)
|
271 |
top_left, bottom_right = box[:2].tolist(), box[2:].tolist()
|
272 |
-
new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple(255, 255, 255), box_pixel+3)
|
273 |
new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple(color) if override_color is None else tuple(override_color), box_pixel)
|
274 |
|
275 |
image = cv2.addWeighted(new_image, box_alpha, image, 1 - box_alpha, 0)
|
|
|
269 |
for box, color in zip(boxes, colors):
|
270 |
box = box.to(torch.int64)
|
271 |
top_left, bottom_right = box[:2].tolist(), box[2:].tolist()
|
272 |
+
new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple((255, 255, 255)), box_pixel+3)
|
273 |
new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple(color) if override_color is None else tuple(override_color), box_pixel)
|
274 |
|
275 |
image = cv2.addWeighted(new_image, box_alpha, image, 1 - box_alpha, 0)
|