Paul Guerrie 5ef69ef3e6 Error in utils/segment/general masks2segments() (#9724) ...

When running segmentation predict on gpu, the conversion from tensor to numpy fails. Calling `.cpu()` solves this problem. 

Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>

Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>

2022-10-06 22:55:15 +02:00

..

aws

Fix yaml.safe_load() ignore emoji errors (#5060)

2021-10-05 13:41:52 -07:00

docker

FROM nvcr.io/nvidia/pytorch:22.09-py3 (#9711)

2022-10-05 22:29:46 +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

Allow list for Comet artifact class 'names' field (#9654)

2022-10-03 00:25:10 +02:00

segment

Error in utils/segment/general masks2segments() (#9724)

2022-10-06 22:55:15 +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

Added cutout import from utils/augmentations.py to use Cutout Aug in … (#9668)

2022-10-01 16:42:31 +02:00

downloads.py

Fix is_url('https://ultralytics.com') (#9610)

2022-09-27 00:29:23 +02:00

general.py

NMS MPS device wrapper (#9620)

2022-09-27 18:02:48 +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