我正在尝试从github运行项目,我正在尝试群集映像,但当我运行项目时,我收到错误ValueError: Error when checking input: expected input_1 to have 4 dimensions, but got array with shape (500, 150528)
。我尝试调试项目,发现这是由这两个函数引起的
def load_images(self):
self.images = []
for image in self.image_paths:
self.images.append(
cv2.cvtColor(cv2.resize(cv2.imread(self.folder_path + "\\" + image), (224, 224)), cv2.COLOR_BGR2RGB))
self.images = np.float32(self.images).reshape(len(self.images), -1)
self.images /= 255
print("\n " + str(
self.max_examples) + " images from the \"" + self.folder_path + "\" folder have been loaded in a random order.")
和
pred = VGG16.predict(self.images)
我不太确定我是否正确地使用了它,或者项目需要一些修改,但是我如何调整代码来预测数组中的图像?
1条答案
按热度按时间aoyhnmkz1#
在第50行中,你提到
VGG16
接受(224,224,3)
形状的输入,但是当你加载图像时,你把它重新整形为(500,150528)
,这就是为什么你会得到一个错误。