我正在尝试构建一个图像生成器,它将:
拍摄原始图像
读入图像并将其大小调整为(224,224,3)以获得resnet50
在其上执行数据扩充(旋转,翻转等)
为此创建一个Resnet50功能(使用model.predict)
然后通过yield方法将其输出到图像生成器中
我不知道是如何实际执行此操作。我将其作为三重损失,因此一个图像是固定图像,正图像和负图像。
它说:
TypeError:len()个未调整大小的对象
更新
我已经更改了一些代码,但是现在它给了我这个错误。
Error when checking input: expected input_1 to have 4 dimensions, but got array with shape (224, 224, 3)
最佳答案
请记住,网络需要一个批量。如果您只提交一个形状为(244,244,3)的图像,则需要先np.expand_dims(image, axis=0)
以获得形状(1,244,244,3),然后再进行预测。那将是1幅244 x 244的图像,具有3层,即(1,244,244,3)。
看来您还有其他一些尺寸,但要点仍然存在,在其中获得该批处理尺寸尺寸。这就是您最近的错误所说的。
关于python - 使用Resnet的ImageGenerator,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57524281/