mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
yolov5/utils
History
Glenn Jocher c4c0ee8fc3
Apple MPS -> CPU NMS fallback strategy (#9600) 
Until more ops are fully supported this update will allow for seamless MPS inference (but slower MPS to CPU transfer before NMS, so slower NMS times).

Partially resolves https://github.com/ultralytics/yolov5/issues/9596

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 		2022-09-26 14:13:03 +02:00
..
aws
…
docker FROM nvcr.io/nvidia/pytorch:22.08-py3 (#9520) 2022-09-20 22:57:42 +02:00
flask_rest_api Allow multiple-model serving from Flask REST API (#8973) 2022-08-15 23:42:25 +02:00
google_app_engine Bump pip from 19.2 to 21.1 in /utils/google_app_engine (#5661) 2021-11-15 20:10:29 +01:00
loggers Update Comet links (#9587) 2022-09-25 20:12:57 +02:00
segment Add segment line predictions (#9571) 2022-09-25 14:52:49 +02:00
__init__.py Standardize warnings with `WARNING ⚠️ ...` (#9467) 2022-09-18 16:15:25 +02:00
activations.py [pre-commit.ci] pre-commit suggestions (#7279) 2022-04-04 22:47:00 +02:00
augmentations.py Update albumentations (#9503) 2022-09-21 19:53:26 +02:00
autoanchor.py Standardize warnings with `WARNING ⚠️ ...` (#9467) 2022-09-18 16:15:25 +02:00
autobatch.py AutoBatch report include reserved+allocated (#9491) 2022-09-19 12:18:55 +02:00
callbacks.py Daemon `plot_labels()` for faster start (#9057) 2022-08-21 03:47:37 +02:00
dataloaders.py generator seed fix for DDP mAP drop (#9545) 2022-09-24 19:35:07 +02:00
downloads.py Update default GitHub assets (#9573) 2022-09-25 02:59:25 +02:00
general.py Apple MPS -> CPU NMS fallback strategy (#9600) 2022-09-26 14:13:03 +02:00
loss.py Fix torch `long` to `float` tensor on HUB macOS (#8067) 2022-06-01 17:34:46 +02:00
metrics.py Standardize warnings with `WARNING ⚠️ ...` (#9467) 2022-09-18 16:15:25 +02:00
plots.py New `scale_segments()` function (#9570) 2022-09-24 16:02:41 +02:00
torch_utils.py Standardize warnings with `WARNING ⚠️ ...` (#9467) 2022-09-18 16:15:25 +02:00
triton.py Detect.py supports running against a Triton container (#9228) 2022-09-24 00:56:42 +02:00