yolov5/models
Zigarss 20d45aa4f1
Add feature map visualization (#3804)
* Add feature map visualization

Add a feature_visualization function to visualize the mid feature map of the model.

* Update yolo.py

* remove boolean from forward and reorder if statement

* remove print from forward

* General cleanup

* Indent

* Update plots.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-28 13:18:45 +02:00
..
hub Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Parameterize max_det + inference default at 1000 (#3215) 2021-05-17 22:47:44 +02:00
experimental.py Merge `develop` branch into `master` (#3518) 2021-06-08 10:22:10 +02:00
yolo.py Add feature map visualization (#3804) 2021-06-28 13:18:45 +02:00
yolov5l.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5m.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5s.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5x.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00