python - 根据路径列表导入图像

Question

所以我正在尝试导入一些图像。所有图像在 .csv 文件的第一列中都有各自的文件路径。以下是我迄今为止采取的方法。这个想法是制作路径列表，然后使用 opencv 将图像“读取”到列表“cv_img”

当前的问题是只有 1 个图像被添加到列表 cv_img 中。知道为什么会发生这种情况吗？

这是最好的方法吗？

#Import Labels
import pandas as pd
df = pd.read_csv('File path.csv',usecols=[3])
labels=df

#Import features
#Problem area
cv_img = []
list1=pd.read_csv('File path.csv',usecols=[0])
for img in list1:
    n= cv2.imread(img)
    cv_img.append(n)

X_train=cv_img
y_train=labels
print(len(X_train)) #prints 1
print(len(y_train)) #prints 1136

Answer 1

我认为 for 循环中存在错误。我认为您的 for 循环只是读取列表中的第一个图像。

尝试将其修改为以下内容:

import cv2
for img in range(len(list1)):
    n= cv2.imread(list1[img])
    cv_img.append(n)