AAAAAAyq commited on
Commit
1e39555
·
1 Parent(s): 7f6d4e3

Fix the OOM from the useful suggestion by hysts

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def fast_process(annotations, image, high_quality, device):
57
  contours, _ = cv2.findContours(annotation, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
58
  for contour in contours:
59
  contour_all.append(contour)
60
- cv2.drawContours(temp, contour_all, -1, (255, 255, 255), 3)
61
  color = np.array([0 / 255, 0 / 255, 255 / 255, 0.9])
62
  contour_mask = temp / 255 * color.reshape(1, 1, -1)
63
  # plt.imshow(contour_mask)
 
57
  contours, _ = cv2.findContours(annotation, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
58
  for contour in contours:
59
  contour_all.append(contour)
60
+ cv2.drawContours(temp, contour_all, -1, (255, 255, 255), 2)
61
  color = np.array([0 / 255, 0 / 255, 255 / 255, 0.9])
62
  contour_mask = temp / 255 * color.reshape(1, 1, -1)
63
  # plt.imshow(contour_mask)