zdou0830 commited on
Commit
a183097
1 Parent(s): d353566
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)