Deeplabcut: Restoring Model and Activation Maps (Function to save scoremaps in dictionary added)

Created on 17 Oct 2018  路  4Comments  路  Source: DeepLabCut/DeepLabCut

Good day,
I would like to know if it is possible to generate activation maps for the models/snapshots generated by train.py? I would like to visualize which particular areas of an image for each particular body part observed (e.g. nose, eyes, hands, feet).

Thank you very much.

All 4 comments

This is perfect! Thank you very much.

Voila: https://github.com/AlexEMG/DeepLabCut/blob/master/Analysis-tools/AnalyzeVideosAndExtractScoreMaps.py

Sorry for the dumb question, but why is the shape for scmap and locref (90,160, 17) and (90, 160, 17, 2) respectively? I understand that the axis 3 value 17 correspond to 17 features that I have used for my model, but I can't figure out why there is 90, 160, and an additional axis for locref with size of 2. For reference, I am analyzing a video with 376 detected frames @25fps, and a resolution of 1280 x 720. Thank you.

90 x 160 are the downsampled dimensions of the input image (transposed). So scmap[:,:,0] is the scoremap for feature 0...

Was this page helpful?
0 / 5 - 0 ratings