打开cv无法读取所有文件形成的路径

Question

我正在处理一个由72个图像和72个掩码组成的数据集。我将图像附加到numpy ndarray中，我希望cv2从与numpy ndarray中的文件相对应的路径中读取文件。

这是通往图像和遮罩的路径：

images_dir = '/content/drive/MyDrive/dataset/images'
masks_dir = '/content/drive/MyDrive/dataset/masks'

#将图像添加到numpy ndarray

file_names = np.sort(os.listdir(images_dir)) 
file_names = np.char.split(file_names, '.')
filenames = np.array([])
for i in range(len(file_names)):
filenames = np.append(filenames, file_names[i][0])

这是我想要打开cv读取每个图像的函数，然后从相应的路径屏蔽：

def augment_dataset(count):
'''Function for data augmentation
Input:
count - total no. of images after augmentation = initial no. of images * count
Output:
writes augmented images (input images & segmentation masks) to the working directory
'''
transform_1 = augment(512, 512)
transform_2 = augment(480, 480)
transform_3 = augment(512, 512)
transform_4 = augment(800, 800)
transform_5 = augment(1024, 1024)
transform_6 = augment(800, 800)
transform_7 = augment(1600, 1600)
transform_8 = augment(1920, 1280)

i = 0
for i in range(count):
for file in filenames:
tile = file.split('_')[1]
img = cv2.imread(images_dir+file+'.jpg')
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
mask = cv2.imread(masks_dir+file+'.png')
mask = cv2.cvtColor(mask, cv2.COLOR_BGR2RGB)

当我运行代码时：

augment_dataset(8)

出现以下错误：

---------------------------------------------------------------------------
error                                     Traceback (most recent call last)
<ipython-input-112-fae4beb79e15> in <module>()
----> 1 augment_dataset(8)
<ipython-input-111-121d55acd3fc> in augment_dataset(count)
20             tile = file.split('_')[1]
21             img = cv2.imread(images_dir+file+'.jpg')
---> 22             img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
23             mask = cv2.imread(masks_dir+file+'.png')
24             mask = cv2.cvtColor(mask, cv2.COLOR_BGR2RGB)
error: OpenCV(4.1.2) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

我知道这是因为OpenCV没有读取这些文件。那么我如何制作openCV来读取文件呢？

Answer 1

在这些情况下，最好使用print(path/to/directory)查看目录是否正确。在这种情况下，我们可以看到我在目录的路径中错过了一个/。因此Python无法解析数据。此外，您可以使用ose.path.exists(path/to/directory)查看路径是否存在。如果返回的值是False，则必须检查指定路径中的错误