【发布时间】:2020-01-07 13:00:12
【问题描述】:
我正在使用 Haralick 纹理来解决医学图像分类问题。我想遍历带有未标记图像的测试文件夹,并将它们打印在带有预测标签的 jupyter 笔记本中。
cv2.imshow() 将输出一个随机图像来显示,但是,当我使用plt.imshow() 在 jupyter 笔记本中显示时,返回的是相同的图像。
# loop over the test images
test_path = 'data/test/test'
for file in glob.glob(test_path + "/*.jpeg"):
# read the input image
image = cv2.imread(file)
# convert to grayscale
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# extract haralick texture from the image
features = extract_features(gray)
# evaluate the model and predict label
prediction = clf_svm.predict(features.reshape(1, -1))[0]
# show the label
cv2.putText(image, prediction, (20,30), cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0,255,255), 3)
# display the output image
#cv2.imshow("Test_Image", image)
#cv2.waitKey(0)
# display the output image in notebook
plt.imshow(image)
plt.show()
使用 pyplot 返回相同的图像,我想返回测试文件夹中的所有(或随机子集)图像。
如果有什么不清楚的,我会澄清的。谢谢。
【问题讨论】:
标签: python opencv jupyter-notebook textures feature-extraction