在Python中使用astype（'float32'）时出错

Question

我是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.

谁能帮我解决这个问题？

Answer 1

遍历您的列表：

在代码末尾，您似乎正在尝试更改存储在列表中的图像的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]