当我尝试导入ResNet50时,出现以下错误:
ValueError:两个形状的尺寸0必须相等,但必须为1并且 512.形状为[1,1,128,512]和[512,256,1,1]。对于输入形状为[1,1,128,512],[512,256,1,1]的'Assign_358'(op:'Assign')。
这是我的代码:
model = ResNet50(weights=None)
model.load_weights('resnet50_weights_tf_dim_ordering_tf_kernels.h5')