Skip to content

Commit

Permalink
Readme fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
makseq committed Aug 23, 2024
1 parent e832c58 commit cca8a58
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 1 deletion.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,8 @@ Check the **Required parameters** column to see if you need to set any additiona
| [spacy](/label_studio_ml/examples/spacy) | NER by [SpaCy](https://spacy.io/) |||| None | Set [(see documentation)](https://spacy.io/usage/linguistic-features) |
| [tesseract](/label_studio_ml/examples/tesseract) | Interactive OCR. [Details](https://github.com/tesseract-ocr/tesseract) |||| None | Set (characters) |
| [watsonX](/label_studio_ml/exampels/watsonx)| LLM inference with [WatsonX](https://www.ibm.com/products/watsonx-ai) and integration with [WatsonX.data](watsonx.data)|||| None| Arbitrary|
| [yolo](/label_studio_ml/examples/yolo) | Object detection with [YOLO](https://docs.ultralytics.com/tasks/) |||| None | Arbitrary |

# (Advanced usage) Develop your model

To start developing your own ML backend, follow the instructions below.
Expand Down
18 changes: 17 additions & 1 deletion label_studio_ml/examples/yolo/README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,22 @@
# YOLO ML Backend for Label Studio

This guide describes the simplest way to start using YOLO ML backend with Label Studio.
The YOLO ML Backend for Label Studio is designed to integrate advanced object detection,
segmentation, classification, and video object tracking capabilities directly into Label Studio.
This integration allows you to leverage powerful YOLO models for various machine learning tasks,
making it easier to annotate large datasets and ensure high-quality predictions.

**Supported Features**

| YOLO Task Name | Label Studio Control Tag | Prediction Supported | Training Supported |
|---------------------------------------|--------------------------------------|----------------------|--------------------|
| Object Detection | `<RectangleLabels>` |||
| Oriented Bounding Boxes (OBB) | `<RectangleLabels model_obb="true">` |||
| Image Instance Segmentation: Polygons | `<PolygonLabels>` |||
| Image Semantic Segmentation: Masks | `<BrushLabels>` |||
| Image Classification | `<Choices>` |||
| Pose Detection | `<KeyPoints>` |||
| Video Object Tracking | `<VideoRectangle>` |||
| Video Temporal Classification | `<Timeline>` | Coming soon ||

! TODO: insert video with demo:
! there should be a video with all supported control tags: RectangleLabels, PolygonLabels, Choices, VideoRectangle.
Expand Down

0 comments on commit cca8a58

Please sign in to comment.