我正在尝试使用 Google Colab 使用 U-net 进行图像分割过程。我可以将图像数据集从 Drive 读取到 Colab 并保存在数组中。仅供参考:我的 Google Drive 中有一个文件夹,其中包含所有训练数据,其中包含 2 个子文件夹(分别为 Image 和 Mask)。
现在,当我使用“plt.show”检查图像时,在阅读并调整图像和蒙版大小后,我注意到图像编号的顺序存在差异。例如,当我随机选取第 10 张图像时,该图像与谷歌驱动器中的第 10 张图像不匹配。更糟糕的是,我的蒙版得到了完全不同的图像,这使得我的图像和蒙版不同(主要问题!!)。
有人遇到过类似的情况吗?知道如何解决这个问题吗?