时间戳

首页 > 解决方案 > 如何从图像测试文件夹中随机显示测试图像

python - 如何从图像测试文件夹中随机显示测试图像

问题描述

我正在使用 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 返回相同的图像，我想从测试文件夹返回所有（或随机子集）图像。

如果有什么不清楚的，我会澄清的。谢谢你。

示例图像

标签： pythonopencvjupyter-notebooktexturesfeature-extraction

解决方案

核心问题是你的循环。您没有在循环期间制表 - 您正在使用一个变量。循环访问的最后一项将基于image、gray和。features

然后在这些单个项目的循环之外完成额外的处理，就像输出一样。

您可能希望将所有处理都带入 for 循环，或者您希望将项目存储在某种列表中，然后对循环项目进行进一步处理，可能会暂停以查看不同的输出。

推荐阅读