Hello, I’m doing a work to classify some problems of conveyor belt structures using images.
For now, I’m trying to identify the accumulation of dirt on the structutres just with RGB images. I’m training a network using transfer learning from vgg16. My dataset is short, yet. But I’m applying the data augmentation.
I would like to see what my network is recognizing, but I don’t know how to visualize the intermediate activitions of my network and create images from them. Is there some method of pytorch to do that or I have to create a new code from scratch?