zhigangjiang commited on
Commit
f9e6ff7
1 Parent(s): a5a934b

Update utils/visibility_polygon.py

Browse files
Files changed (1) hide show
  1. utils/visibility_polygon.py +4 -4
utils/visibility_polygon.py CHANGED
@@ -230,13 +230,13 @@ def visualization(segments: np.array, output_polygon: np.array, center: np.array
230
  board = np.zeros((side_l, side_l))
231
  for segment in segments:
232
  segment = (segment - origin) * scale + origin
233
- segment = segment.astype(np.int)
234
  cv2.line(board, tuple(segment[0]), tuple(segment[1]), 0.5, thickness=3)
235
- board = cv2.drawMarker(board, tuple(origin.astype(np.int)), 1, thickness=3)
236
 
237
  output_polygon = (output_polygon - origin) * scale + origin
238
- board = cv2.drawContours(board, [output_polygon.astype(np.int)], 0, 1, 3)
239
- board = cv2.drawMarker(board, tuple(origin.astype(np.int)), 1, thickness=3)
240
  plt.axis('off')
241
  plt.imshow(board)
242
  plt.show()
 
230
  board = np.zeros((side_l, side_l))
231
  for segment in segments:
232
  segment = (segment - origin) * scale + origin
233
+ segment = segment.astype(np.int32)
234
  cv2.line(board, tuple(segment[0]), tuple(segment[1]), 0.5, thickness=3)
235
+ board = cv2.drawMarker(board, tuple(origin.astype(np.int32)), 1, thickness=3)
236
 
237
  output_polygon = (output_polygon - origin) * scale + origin
238
+ board = cv2.drawContours(board, [output_polygon.astype(np.int32)], 0, 1, 3)
239
+ board = cv2.drawMarker(board, tuple(origin.astype(np.int32)), 1, thickness=3)
240
  plt.axis('off')
241
  plt.imshow(board)
242
  plt.show()