Как эффективно загрузить много изображений из папки, используя openCV

Question

Я пытаюсь создать свои собственные наборы данных изображений для машинного обучения.

Рабочий процесс, который я думал, следующий:

oЗагрузить все файлы изображений в виде массива в папке.

abПомечать загруженные изображения

p Разделить загруженные файлы изображений на image_data и label_data.

in Наконец, разделите image_data на image_train_data и image_test_data и разделите label_data на label_train_data и label_test_data.

Однако на первом этапе все идет не так (①).

Как эффективно загрузить все данные изображения?

А если вы реализуете набор изображений для машинного обучения в соответствии с этим рабочим процессом, как вы справитесь с этим?

Я написал следующий код.

cat_im = cv2.imread("C:\\Users\\path\\cat1.jpg")

Но, я вынужден писать \ cat1.jpg, \ cat2.jpg, \ cat3.jpg .....?

Answer 1

## you can find all images like extenstion
import os,cv2
import glob
all_images_path= glob.glob('some_folder\images\*png') ## it gives path of images as list   

## then you can loop over all files
loaded_images  = []
for image_path in all_images_path:
    image = cv2.imread(image_path)
    loaded_images.append(image)

## lets assume your labels are just name of files and its like cat1.png,cat2.png etc
labels = []
for image_path in all_images_path:
    labels.append(os.basename(image_path))