Get Features maps


I am working with Nvidia TAO for training my models.

I discovered the Nvidia tool NVIDIA Feature Map Explorer which needs some .npy files containing the features maps.

How could I get the features maps in .npy files of my trained model in TAO ?

Thanks you for your help,

Is it the same as Get features maps - #5 by Morganh ?

yes, excuse me for the double topic

No problem. Let us follow up with this topic.

TAO does not provide an API to dump the numpy array for the feature map tensors. So currently it is not feasible to visualize layer’s feature map with NVIDIA FME.

NVIDIA Feature Map Explorer

Not surprisingly, in order to use Feature Map Explorer, you will need to generate some feature maps from your trained model. This is a straightforward task with most training frameworks, since FME directly consumes feature map tensors written out as standard numpy array files (i.e., the .npy files written by the method). These files are assumed to contain 32-bit single precision floating point values in a 4D array with order NCHW (where N is the batch size, C is the channel count, and H and W are height and width respectively). If your training data is in NHWC (channels last) layout, you will need to transpose the axes (e.g., permute the axes with a ( 0 , 3 , 1 , 2 ) permutation) before calling .

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.