Relative Content

Tag Archive for pythonpytorchobject-detectionsingle-shot-detector

How to visualize feature maps for SSD model

I’m developing a Single Shot Multibox Detection model with MobileNet as the backbone with PyTorch. I’d like to visualize the feature maps to help debugging and ensure that the feature map is picking up what I need it to see. I have a model loaded which I know works (I tested real predictions), and am trying to use TorchCam to visualize the last convolutional layer of the SSD Backbone.