Hi everyone.
I am working with pytorch, and trying to implement a semantic segmentation method.
I want to visualize a feature map whose size is [10,192,128,128], 10 is the batch size, 192 is the number of channels and (128,128) being spatial dimension.
Is there anyway to visualise this feature map using tensorboard in pytorch.??
I tried using add_image(), but I think it works only with 3 channels.