update

Doodleverse · Feb 11, 2023 · 95bb717 · 95bb717
1 parent 1e26ef8
commit 95bb717
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/doodleverse_utils/prediction_imports.py b/doodleverse_utils/prediction_imports.py
@@ -440,22 +440,22 @@ def do_seg(
         plt.imshow(bigimage, cmap='gray')
     else:
         plt.imshow(bigimage[:, :, :3])
-    if NCLASSES>2:
-        plt.imshow(color_label, alpha=0.5)
-    elif NCLASSES==2:
-        cs = plt.contour(est_label, [-99,0,99], colors='r')
+    # if NCLASSES>2:
+    plt.imshow(color_label, alpha=0.5)
+    # elif NCLASSES==2:
+    #     cs = plt.contour(est_label, [-99,0,99], colors='r')
     plt.axis("off")
     # plt.show()
     plt.savefig(segfile, dpi=200, bbox_inches="tight")
     plt.close("all")
 
-    if NCLASSES==2:
-        segfile = segfile.replace("_overlay.png", "_result.mat")
-        p = cs.collections[0].get_paths()[0]
-        v = p.vertices
-        x = v[:,0]
-        y = v[:,1]
-        io.savemat(segfile, dict(x=x, y=y))
+    # if NCLASSES==2:
+    #     segfile = segfile.replace("_overlay.png", "_result.mat")
+    #     p = cs.collections[0].get_paths()[0]
+    #     v = p.vertices
+    #     x = v[:,0]
+    #     y = v[:,1]
+    #     io.savemat(segfile, dict(x=x, y=y))
 
 # --------------------------------------------------------
 def make_gradcam_heatmap(image, model):