Spaces:
Runtime error
Runtime error
AAAAAAyq
commited on
Commit
·
1e39555
1
Parent(s):
7f6d4e3
Fix the OOM from the useful suggestion by hysts
Browse files
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),
|
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)
|