Hello,
As you can see in the image I get 3 tensors of the imagen, how can I merge it as the image show above?
Thanks
I think u hv to index into the 1st element which is the predicted mask upload[0]
To superimpose you can see the doc here
the code will look something like this where img is PILImage object while msk is PILMask object
_,axs = plt.subplots(1,1, figsize=(8,8))
img.show(ctx=axs)
msk.show(ctx=axs)
See the inference section from my lesson notebook:
I cover this
1 Like