Commit Graph
3 Commits
Author SHA1 Message Date
6f377f9d8a Feature/classification tutorial refactor (#10039)
* Added a tutorial notebook for classification.

* Split a cell so that there is less room for error when a user pastes their own code snippet. Also added an active learning section at the end.

* Added a section to the classification tutorial notebook about the various methods of input for `classify/predict.py`.

* Updated link to colab

* WIP commit to show some of the errors when trying to match the main tutorial.

* Refactored the classification tutorial to be closer to the main tutorial.

* Update tutorial.ipynb

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

* 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-11-18 22:05:45 +01:00
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