Access to UNet encoder backbone in Metalhead

cirobr
Sun, 14 Sep 2025 19:23:26 +0000
General Usage

Cheers,

In Metalhead, consider the following model:

bb = backbone( ResNet(50) )
model = UNet((256,256), 3, 10, bb)

Need to get access to a specific activation map from the model’s encoder (actually, to a specific activation map in the ResNet backbone). The activation map is easily accessible by choosing the appropriate layer from the “bb” variable, but apparently not from “model”, as Metalhead’s function backbone() seems not to work for the UNet.

Thanks in advance.

1 post - 1 participant