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

..

__init__.py

YOLOv5 segmentation model support (#9052)

2022-09-16 00:12:46 +02:00

augmentations.py

YOLOv5 segmentation model support (#9052)

2022-09-16 00:12:46 +02:00

dataloaders.py

generator seed fix for DDP mAP drop (#9545)

2022-09-24 19:35:07 +02:00

general.py

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

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

loss.py

YOLOv5 segmentation model support (#9052)

2022-09-16 00:12:46 +02:00

metrics.py

YOLOv5 segmentation model support (#9052)

2022-09-16 00:12:46 +02:00

plots.py

Add Paddle exports to benchmarks (#9459)

2022-09-18 00:17:31 +02:00