How to merge TensorMask and TensorImage to show the image


As you can see in the image I get 3 tensors of the imagen, how can I merge it as the image show above?


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))

See the inference section from my lesson notebook:

I cover this

1 Like