嗨,我正在构建一个图像分类器,这是我代码的一小部分
在这一部分中,我尝试将“img_data_list”是一个数组,我正在将该数组转换为 np.array 但是
(img_data = np.array(img_data_list))
这行代码给我一个错误
(无法将输入数组从形状(128,128,3)广播到形状(128,128))我不知道为什么会发生这种情况
for dataset in data_dir_list:
img_list=os.listdir(data_path+'/'+ dataset)
print ('Loaded the images of dataset-'+'{}\n'.format(dataset))
for img in img_list:
image_path = os.path.join(data_path, dataset, img)
input_img=cv2.imread(image_path)
if input_img is not None:
input_img_resize=cv2.resize(input_img,(128,128))
img_data_list.append(input_img_resize)
else:
print(img+' image didnt read')
img_data = np.array(img_data_list)
img_data = img_data.astype('float32')
img_data /= 255
我找到了解决方案,删除数据集中的某些图像后分类器工作正常
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)