From b067ced74fa24a76edd437f26f8daed9ace55290 Mon Sep 17 00:00:00 2001 From: TengjiaoJiang Date: Wed, 28 Jun 2023 14:30:08 +0200 Subject: [PATCH 1/2] This is an example. --- opensd_project/visualize.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/opensd_project/visualize.py b/opensd_project/visualize.py index 15d0b0a..99dcc0c 100644 --- a/opensd_project/visualize.py +++ b/opensd_project/visualize.py @@ -23,6 +23,8 @@ def show_reference_image(images, points=[], roi_size=None, bit_depth=12): if points: ax.scatter(np.array(points)[:, 1], np.array(points)[:, 0], marker='.', color='r') + ax.set_xlabel("xx") + ax.set_ylabel("yy") if roi_size is not None: for point in np.array(points): From be71f0fd3744de8be46e88fbd181068ce08d615c Mon Sep 17 00:00:00 2001 From: TengjiaoJiang Date: Wed, 28 Jun 2023 14:48:35 +0200 Subject: [PATCH 2/2] A new founction of saving images! --- opensd_project/visualize.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/opensd_project/visualize.py b/opensd_project/visualize.py index 99dcc0c..845c9ec 100644 --- a/opensd_project/visualize.py +++ b/opensd_project/visualize.py @@ -2,6 +2,7 @@ import matplotlib.pyplot as plt from matplotlib import animation from matplotlib import patches +import cv2 def show_reference_image(images, points=[], roi_size=None, bit_depth=12): @@ -23,8 +24,6 @@ def show_reference_image(images, points=[], roi_size=None, bit_depth=12): if points: ax.scatter(np.array(points)[:, 1], np.array(points)[:, 0], marker='.', color='r') - ax.set_xlabel("xx") - ax.set_ylabel("yy") if roi_size is not None: for point in np.array(points): @@ -59,3 +58,8 @@ def updatefig(i): ani = animation.FuncAnimation(fig, updatefig, blit=True, frames=images.shape[0], interval=1000/fps) return ani + +def image_save(image,save_path): + filename = 'savedImage.jpg' + cv2.imwrite(save_path+filename,image) + print('Successfully saved') \ No newline at end of file