Access to UNet encoder backbone in Metalhead

General Usage
,
1

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.