Commit Graph
2 Commits
Author SHA1 Message Date
a83d2a5013 Segmentation Tutorial (#9521)
* Added a tutorial notebook for segmentation.

* Updated header for segmentation tutorial and included other YOLOv5 sponsor sections.

* Updated segmentation tutorial to match main object detection tutorial.

* Update tutorial.ipynb

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-31 15:25:11 +01:00
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