Help with generating output/activation map from RetinaNet Model (PyTorch)

Follow the full discussion on Reddit.
Hello everyone. I am fairly new to working with object detectors, and using RetinaNet for now as my choice with PyTorch. I am trying to create an output mapping by feeding an example image into a pre-trained RetinaNet model. I found a research paper that mentions the output mapping containing 8M activations which they simply found by feeding the example image into the model.

Comments

There's unfortunately not much to read here yet...

Discover the Best of Machine Learning.

Ever having issues keeping up with everything that's going on in Machine Learning? That's where we help. We're sending out a weekly digest, highlighting the Best of Machine Learning.

Join over 900 Machine Learning Engineers receiving our weekly digest.

Best of Machine LearningBest of Machine Learning

Discover the best guides, books, papers and news in Machine Learning, once per week.

Twitter