मैं काम कर रहा हूँ के साथ एक डाटासेट के 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 फ़ाइलों को पढ़ने के लिए?