yolov5/utils
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
..
aws Fix datasets for aws and get_coco.sh (#3788) 2021-06-26 15:29:16 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
wandb_logging W&B: Update Tables API and comply with new dataset_check (#3772) 2021-06-25 20:45:33 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
autoanchor.py Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
datasets.py NGA xView 2018 Dataset Auto-Download (#3775) 2021-06-26 00:49:05 +02:00
general.py NGA xView 2018 Dataset Auto-Download (#3775) 2021-06-26 00:49:05 +02:00
google_utils.py Parse URL authentication (#3424) 2021-06-03 18:44:58 +02:00
loss.py Update objectness IoU sort (#3786) 2021-06-26 14:52:18 +02:00
metrics.py ConfusionMatrix `normalize=True` fix (#3587) 2021-06-11 11:46:05 +02:00
plots.py Add feature map visualization (#3804) 2021-06-28 13:18:45 +02:00
torch_utils.py Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00