在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]