我是python的新手,正在使用以下代码:
for index, person in enumerate(people):
print(index)
dir_path = 'train/' + person
for img_path in os.listdir(dir_path):
name, ext = os.path.splitext(img_path)
if ext.lower() not in valid_images:
continue
img_data = cv2.imread(dir_path + '/' + img_path)
# convert image to gray
img_data=cv2.cvtColor(img_data, cv2.COLOR_BGR2GRAY)
img_data_list.append(img_data)
labels.append(index)
img_data = np.array(img_data_list)
img_data = img_data.astype('float32')
但是在运行时出现错误:
img_data = img_data.astype('float32')ValueError: setting an array element with a sequence.
谁能帮我解决这个问题?
最佳答案
遍历您的列表:
在代码末尾,您似乎正在尝试更改存储在列表中的图像的dtype
。用OpenCV读取的图像自然是numpy
数组。
以下示例应有帮助:
# create a test image list
img = np.ones((60,60), dtype=np.uint8)
img_list = [img] * 4
# use a list comp to run through the images and change dtype
changed_img_list = [img.astype(np.float32) for i in img_list]