將圖像饋送到預(yù)訓(xùn)練的 InceptionResNetV2 網(wǎng)絡(luò)時(shí),我得到以下結(jié)果。from keras.applications.inception_resnet_v2 import InceptionResNetV2INPUT_SHAPE = (200, 250, 3)img = load_img() # loads a 200x250 rgb image into a (200, 250, 3) numpy arrayassert img.shape == INPUT_SHAPE # just finemodel = InceptionResNetV2(include_top=False, input_shape=INPUT_SHAPE)model.predict(img)ValueError:檢查輸入時(shí)出錯(cuò):預(yù)期 input_1 有 4 個(gè)維度,但得到了形狀為 (200, 150, 3) 的數(shù)組我不明白模型為什么以及如何期望 4 維輸入。必須做什么來調(diào)整 (200, 250, 3) 圖像以便模型可以處理它?
2 回答

米脂
TA貢獻(xiàn)1836條經(jīng)驗(yàn) 獲得超3個(gè)贊
嘗試使用形狀 (1, 200, 150, 3) 或 (200, 150, 3, 1) 重塑您的輸入。
您可以使用image = np.expand_dims(image, axis=0))
或 image = input_data.reshape((-1, image_side1, image_side2, channels))

萬千封印
TA貢獻(xiàn)1891條經(jīng)驗(yàn) 獲得超3個(gè)贊
您需要提供一批圖像。如果您的批次有一個(gè)圖像,它也應(yīng)該具有相同的格式。
嘗試img.reshape((1, 200, 150, 3))
添加回答
舉報(bào)
0/150
提交
取消