From 95bb71727b573e7ac1f844b2724de240b29114d9 Mon Sep 17 00:00:00 2001 From: dbuscombe-usgs Date: Fri, 10 Feb 2023 20:17:44 -0800 Subject: [PATCH] update --- doodleverse_utils/prediction_imports.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/doodleverse_utils/prediction_imports.py b/doodleverse_utils/prediction_imports.py index ae0b109..5694274 100755 --- 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):