如何使用openCV從文件夾中有效地加載許多圖像

Question

我嘗試為機器學習創建自己的圖像數據集。

我認為的工作流程如下：

①將所有圖像文件以數組形式加載到文件夾中。

②標記加載的圖像

③將加載的圖片文件拆分為image_data和label_data。

④最后將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))