如何在keras中预测多个图像

wsxa1bj1  于 2022-11-24  发布在  其他
关注(0)|答案(1)|浏览(164)

我正在尝试从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)

我不太确定我是否正确地使用了它,或者项目需要一些修改,但是我如何调整代码来预测数组中的图像?

aoyhnmkz

aoyhnmkz1#

在第50行中,你提到VGG16接受(224,224,3)形状的输入,但是当你加载图像时,你把它重新整形为(500,150528),这就是为什么你会得到一个错误。

self.images = np.float32(self.images).reshape(len(self.images), 224,224,3)

相关问题