Hello My input is [batch, H, W, C]. I see that the visualization only output result for first element in the batch, like [0, H, W, C]. Is there a way to specify which element to visualize ?? Best He Zhao
Hello
My input is [batch, H, W, C]. I see that the visualization only output result for first element in the batch, like [0, H, W, C].
Is there a way to specify which element to visualize ??
Best
He Zhao