代码复现结果极差的问题
Created by: lxk767363331
在复现R2UNet时结果极差不知道哪里出了问题，从结构上看 我应该没有浮现错误。下面分别是我的代码与torch代码
‘’ def RRCNN_block(data,out_ch): with scope("conv0"): x =conv(data, out_ch, 1, stride=1, padding=0) with scope("Recurrent_block1"): data=Recurrent_block(x,out_ch) data=Recurrent_block(data,out_ch) return x+data
def Recurrent_block(data,out_ch): with scope("rconv0"): x1 = bn_relu( conv(data, out_ch, 3, stride=1, padding=1)) with scope("rconv1"): r1=data+x1 r1 = bn_relu( conv(r1, out_ch, 3, stride=1, padding=1)) with scope("rconv2"): r2=r1+data r2 = bn_relu( conv(r2, out_ch, 3, stride=1, padding=1)) return r2 ‘’
我是讲unet代码中的所有double_conv都替换为RRCNN_block。 在眼底数据集上，目前miou效果最佳没有超过0.85，unet的miou效果是0.91左右。复现的attention unet效果是可以的。