Advertisement
Guest User

Untitled

a guest
Jun 20th, 2018
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.80 KB | None | 0 0
  1. def draw_contours_on_image(contours, img):
  2. color = SCALAR_GREEN
  3. for contour in contours:
  4. cv2.polylines(img, [contour], True, color=color, thickness=2)
  5.  
  6.  
  7. def draw_boxes_on_image(boxes, img):
  8. for box in boxes:
  9. color = (colours[box.track_id % 11], colours[box.track_id * 23 % 13], colours[box.track_id * 11 % 17])
  10. p1 = (int(box.left), int(box.bottom))
  11. p2 = (int(box.right), int(box.top))
  12. if box.phantom:
  13. drawrect(img, p1, p2, color=color, thickness=2, style='dashed')
  14. else:
  15. cv2.rectangle(img, p1, p2, color=color, thickness=2)
  16.  
  17. # dblFontScale = blobs[i].dblCurrentDiagonalSize / 60.0
  18. # intFontThickness = int(dblFontScale * 1.0)
  19.  
  20. # cv2.putText(imgFrame2, str(i), blobs[i].centerPositions[-1], cv2.FONT_HERSHEY_COMPLEX, dblFontScale,
  21. # SCALAR_GREEN,
  22. # intFontThickness)
  23.  
  24.  
  25. def draw_boxes_with_contour(contours, boxes, img_con):
  26. for box in boxes:
  27. if not box.phantom:
  28. mid_y = (box.left + box.right) / 2
  29. mid_x = (box.top + box.bottom) / 2
  30. mid_p = (mid_y, mid_x)
  31.  
  32. p1 = (int(box.left), int(box.bottom))
  33. p2 = (int(box.right), int(box.top))
  34. for i, contour in enumerate(contours):
  35. res = cv2.pointPolygonTest(contour, mid_p, False)
  36. if res == OUTSIDE_CONTOUR:
  37. cv2.rectangle(img_con, p1, p2, color=SCALAR_RED, thickness=2)
  38. else:
  39. cv2.rectangle(img_con, p1, p2, color=SCALAR_GREEN, thickness=2)
  40. else:
  41. p1 = (int(box.left), int(box.bottom))
  42. p2 = (int(box.right), int(box.top))
  43. cv2.rectangle(img_con, p1, p2, color=SCALAR_RED, thickness=2)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement