numpy - 图像和蒙版可视化 - Colab
问题描述
大家好。
我已经使用 matplotlib、NumPy 和 Random 在 Colab 中设置了图像和蒙版视觉检查,并且预计会为两者(图像和蒙版)打印相同的图像编号,但不幸的是,事实并非如此。无论出于何种原因,图像不对应。两个图像具有相同的大小和编号/名称。有人对如何解决这个问题有一些提示吗?先感谢您!
import random
import numpy as np
image_number = random.randint(0,len(image_dataset))
plt.figure(figsize=(12,6))
plt.subplot(121)
plt.imshow(np.reshape(image_dataset[image_number], (patch_size,patch_size,3)))
plt.subplot(122)
plt.imshow(np.reshape(mask_dataset[image_number], (patch_size,patch_size,3)))
plt.show()
这就是我导入训练图像的方式。(对面具做同样的事情)
images_dataset = [] #TRAIN IMAGES
for path, subdirs, files in os.walk(root_directory):
dirname = path.split(os.path.sep)[-1]
if dirname =="images":
images = os.listdir(path)
for i, image_name in enumerate(images):
if image_name.endswith('.png'):
image = cv2.imread(path + "/" + image_name, 1)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
SIZE_X = (image.shape[1]//patch_size)*patch_size
SIZE_Y = (image.shape[0]//patch_size)*patch_size
image = Image.fromarray(image)
image = image.crop((0,0,SIZE_X, SIZE_Y))
image = np.array(image)
解决方案
尝试整理数据。
images = sorted(os.listdir(path))
推荐阅读
- visual-studio-code - 在 Visual Studio Code 的 Latex 研讨会中启用 xelatex
- pandas - 如果数据框已经删除了缺失的行,如何检测缺失值?
- python - 包含列表python输出的html下拉列表
- android - AWS AppSync 错误 - 空对象引用上的 AWSMobileClientStore.get(String)
- gatsby - 如何通过 gatsby-transformer-asciidoc 从本地目录获取图像文件?
- python-3.x - 按日期分组到星期
- php - Docker配置虚拟域名错误,无法访问
- opendaylight - 醒酒器框架在 OpenDaylight Karaf shell 中无法按预期工作
- jquery - 自动完成选择名称并将 id 插入隐藏输入
- unit-testing - 在 Jest 中模拟模块时如何断言默认导出函数被调用?