我想在colab中使用cv2_imshow

Question

import cv2 as cv
import numpy as np
from google.colab.patches import cv2_imshow
from tensorflow.keras.models import load_model

img_color = cv.imread('test3.jpg', cv.IMREAD_COLOR)
img_gray = cv.cvtColor(img_color, cv.COLOR_BGR2GRAY)

ret, img_binary = cv.threshold(img_gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)

kernel = cv.getStructuringElement(cv.MORPH_RECT, (5, 5))
img_binary = cv.morphologyEx(img_binary, cv.MORPH_CLOSE, kernel)

cv2_imshow('digit', img_binary)
cv.waitKey(0)

这是手写训练模型并测试训练好的模型的过程。 我想在 colab 中使用 imshow() 加载图像。

有没有办法不用设置文件路径就可以使用？

Answer 1

您可以使用matplotlib的函数来实现

import matplotlib.pyplot as plt
%matplotlib inline

fig = plt.gcf()
fig.set_size_inches(18, 10)
plt.axis("off")
plt.rcParams['figure.figsize'] = [20, 10]
plt.imshow(img_binary)
plt.show()