擅长:python、mysql、java
<p>实现所需功能的一种简单方法是将文件名作为参数传递给函数,并在函数内部打开图像文件。这样,您就可以使用文件名字符串来清理它并将其设置为图像标题。您还可以添加一个参数以具有单个打印功能,并选择是要打印彩色图像还是灰度图像:</p>
<pre><code>import os
def plotImage(file, grayscale=False):
image = cv2.imread(file)
filename = os.path.basename(file).split('.')[0]
plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2GRAY if grayscale else cv2.COLOR_BGR2RGB))
plt.title(filename)
plotImage("lena.png") # plot color image
plotImage("link.png", grayscale=True) # plot grayscale image
</code></pre>