mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
yolov5/utils/segment
History
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