Using Unet with Densenet

(jaideep v) #1

can any one suggest how we can use Densenet with Unet.
This is for the resnet34 ,i want to use Densnet instead of Resnet34. Resnet34 has 7 block but densenet has only 0 and many sublocks in it.

self.sfs = [SaveFeatures(rn[i]) for i in [2,4,5,6]]
self.up1 = UnetBlock(512,256,256)
self.up2 = UnetBlock(256,128,256)
self.up3 = UnetBlock(256,64,256)
self.up4 = UnetBlock(256,64,256)
self.up5 = nn.ConvTranspose2d(256, 1, 2, stride=2)

0 Likes

#2

Checkout Tiramisu

0 Likes