Need help aggregating the feature maps into smaller data

[rocketsfallonrocketfalls]

Can anyone help me understand what are the dimensions of the feature vectors explicitly for and how can I use them for my application?

My actual problem:

• I want to calculate similarity based on image features calculated by the network.
• Using the features I obtain using yolov5m cause computational problems as I have to load these vectors for hundreds of images and perform operations on them.

In my application in particular, I get 3 sets of feature maps for a single image flowing into the Detect layer. Their dimensions are:

1. 192 x 32 x 40
2. 384 x 16 x 20
3. 768 x 8 x 10

My actual question:

• How can I downsize/dimensionally-reduce these data for similarity checking purposes?

[glenn-jocher]

@rocketsfallonrocketfalls you may want to see AdaptiveAvgPool2d for averaging over the last 2 spatial dimensions.

https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool2d.html

[glenn-jocher]

@rocketsfallonrocketfalls good news 😃! Feature map visualization was added ✅ in PR #3804 by @Zigars today. This allows for visualizing feature maps from any part of the model from any function (i.e. detect.py, train.py, test.py). Feature maps are saved as *.png files in runs/features/exp directory. To turn on feature visualization set feature_vis=True in the model forward method and define the layer you want to visualize (default is SPP layer).

yolov5/models/yolo.py

Lines 158 to 160 in 20d45aa

	if feature_vis and m.type == 'models.common.SPP':
	feature_visualization(x, m.type, m.i)

To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload with model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

[rocketsfallonrocketfalls]

@glenn-jocher, @Zigars, that's great news, thanks a lot!

[rocketsfallonrocketfalls]

2D Pooling seems to work fine for now :)