39 lines
864 B
Python
39 lines
864 B
Python
import cv2
|
|
from PIL import Image
|
|
from matplotlib import pyplot as plt
|
|
|
|
|
|
def show_image_cv2(image, maxdim=700):
|
|
h, w, c = image.shape
|
|
maxhw = max(h, w)
|
|
if maxhw > maxdim:
|
|
ratio = maxdim / maxhw
|
|
h = int(h * ratio)
|
|
w = int(w * ratio)
|
|
img = cv2.resize(image, (h, w))
|
|
cv2.imshow("", img)
|
|
cv2.waitKey(0)
|
|
cv2.destroyAllWindows()
|
|
|
|
|
|
def show_image_mpl(image):
|
|
fig, ax = plt.subplots(1, 1)
|
|
fig.set_size_inches(20, 20)
|
|
ax.imshow(image, cmap="gray")
|
|
plt.show()
|
|
|
|
|
|
def show_image(image, backend="mpl"):
|
|
if backend == "mpl":
|
|
show_image_mpl(image)
|
|
elif backend == "cv2":
|
|
show_image_cv2(image)
|
|
elif backend == "pil":
|
|
Image.fromarray(image).show()
|
|
else:
|
|
raise ValueError(f"Unknown backend: {backend}")
|
|
|
|
|
|
def save_image(image, path):
|
|
cv2.imwrite(path, image)
|