我在Keras中定义了Double conv层,将其用作Unet体系结构中的参考,如下所示:
class ConvBlock(keras.Model):
def __init__(self,in_channels):
super(ConvBlock,self).__init__()
self.in_channels = in_channels
self.conv = keras.Sequential(
layers.Conv2D(filters=self.in_channels,kernel_size=(3,3),strides=(1,1),padding="same",use_bias=False),
# layers.BatchNormalization(),
# layers.ReLU(),
layers.Conv2D(filters=self.in_channels,kernel_size=(3,3),strides=(1,1),padding="same", use_bias=False),
# layers.BatchNormalization()
# layers.ReLU(),
)
def call(self,input_tensor):
return self.conv(input_tensor)
在单元测试期间,我遇到了以下错误:ValueError:name\u范围的名称必须是字符串。 这是我的代码测试,非常简单
model = ConvBlock(64)
y = tf.random.uniform((2,32,32,3))
print(model(y).shape)
tf.keras.keras.Sequential
获取图层列表,请尝试以下操作:相关问题 更多 >
编程相关推荐