Metalhead.UNet output array limit

cirobr
Sat, 10 Feb 2024 13:07:06 +0000
General Usage

Cheers,

Would like to confirm if the yhat array output of Metalhead.UNet model is limited to [0:1] elements. Apparently its upper bound is higher than 1.

Model is instantiated as follows:

backbone = Metalhead.backbone( ResNet(50; pretrain=true) )
modelcpu = Metalhead.UNet((512,512), 3, 1, backbone)
model = modelcpu |> gpu;

Then, without training, model is excited with:

x = rand(Float32, (512,512,3,1) |> gpu
yhat = model(x)

Thanks in advance for clarifying.

3 posts - 2 participants