From ca9e2b00fca2326c2321829f164910a06d06c987 Mon Sep 17 00:00:00 2001 From: Max Tkachenko Date: Wed, 18 Sep 2024 18:08:53 +0100 Subject: [PATCH 1/3] Update YOLO_CLASSES.md --- label_studio_ml/examples/yolo/YOLO_CLASSES.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/label_studio_ml/examples/yolo/YOLO_CLASSES.md b/label_studio_ml/examples/yolo/YOLO_CLASSES.md index 49db68a5d..e8cfaade0 100644 --- a/label_studio_ml/examples/yolo/YOLO_CLASSES.md +++ b/label_studio_ml/examples/yolo/YOLO_CLASSES.md @@ -3,7 +3,7 @@ ### Available "yolov8m.pt" model labels for RectangleLabels: ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush'] -### Available "yolov8n-obb.pt" model labels for RectangleLabels.OBB:** +### Available "yolov8n-obb.pt" model labels for RectangleLabels.OBB: ['plane', 'ship', 'storage tank', 'baseball diamond', 'tennis court', 'basketball court', 'ground track field', 'harbor', 'bridge', 'large vehicle', 'small vehicle', 'helicopter', 'roundabout', 'soccer ball field', 'swimming pool'] ### Available "yolov8n-pose.pt" model labels for Keypoints: From dd31cb6dd7bdc66498cfce10e8aab43f446c19f5 Mon Sep 17 00:00:00 2001 From: Max Tkachenko Date: Thu, 19 Sep 2024 12:36:52 +0100 Subject: [PATCH 2/3] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index f5420eac3..4d04931f6 100644 --- a/README.md +++ b/README.md @@ -61,7 +61,7 @@ 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 | +| [yolo](/label_studio_ml/examples/yolo) | All YOLO tasks are supported: [YOLO](https://docs.ultralytics.com/tasks/) | ✅ | ❌ | ❌ | None | Arbitrary | # (Advanced usage) Develop your model From c910e30ec99f791cc3f8bdd3fa139b2af6d18e06 Mon Sep 17 00:00:00 2001 From: Max Tkachenko Date: Thu, 19 Sep 2024 14:39:46 +0100 Subject: [PATCH 3/3] fix: ML backend logger in docker doesn't show errors (#628) --- label_studio_ml/default_configs/_wsgi.py.tmpl | 1 + label_studio_ml/examples/bert_classifier/_wsgi.py | 1 + label_studio_ml/examples/easyocr/_wsgi.py | 1 + label_studio_ml/examples/flair/_wsgi.py | 1 + label_studio_ml/examples/gliner/_wsgi.py | 1 + label_studio_ml/examples/grounding_dino/_wsgi.py | 1 + label_studio_ml/examples/grounding_sam/_wsgi.py | 1 + label_studio_ml/examples/huggingface_llm/_wsgi.py | 1 + label_studio_ml/examples/huggingface_ner/_wsgi.py | 1 + label_studio_ml/examples/interactive_substring_matching/_wsgi.py | 1 + label_studio_ml/examples/llm_interactive/_wsgi.py | 1 + label_studio_ml/examples/mmdetection-3/_wsgi.py | 1 + label_studio_ml/examples/nemo_asr/_wsgi.py | 1 + label_studio_ml/examples/segment_anything_2_image/_wsgi.py | 1 + label_studio_ml/examples/segment_anything_2_video/_wsgi.py | 1 + label_studio_ml/examples/segment_anything_model/_wsgi.py | 1 + label_studio_ml/examples/sklearn_text_classifier/_wsgi.py | 1 + label_studio_ml/examples/spacy/_wsgi.py | 1 + label_studio_ml/examples/tesseract/_wsgi.py | 1 + label_studio_ml/examples/watsonx_llm/data_wsgi.py | 1 + label_studio_ml/examples/watsonx_llm/model_wsgi.py | 1 + label_studio_ml/wsgi.py | 1 + 22 files changed, 22 insertions(+) diff --git a/label_studio_ml/default_configs/_wsgi.py.tmpl b/label_studio_ml/default_configs/_wsgi.py.tmpl index ac2e46b9d..4274a1946 100644 --- a/label_studio_ml/default_configs/_wsgi.py.tmpl +++ b/label_studio_ml/default_configs/_wsgi.py.tmpl @@ -6,6 +6,7 @@ import logging.config logging.config.dictConfig({{ "version": 1, + "disable_existing_loggers": False, "formatters": {{ "standard": {{ "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/bert_classifier/_wsgi.py b/label_studio_ml/examples/bert_classifier/_wsgi.py index 34095d47f..baeec1679 100644 --- a/label_studio_ml/examples/bert_classifier/_wsgi.py +++ b/label_studio_ml/examples/bert_classifier/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/easyocr/_wsgi.py b/label_studio_ml/examples/easyocr/_wsgi.py index 72579b197..389d787bf 100644 --- a/label_studio_ml/examples/easyocr/_wsgi.py +++ b/label_studio_ml/examples/easyocr/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/flair/_wsgi.py b/label_studio_ml/examples/flair/_wsgi.py index c7eb0f029..fdf40b306 100644 --- a/label_studio_ml/examples/flair/_wsgi.py +++ b/label_studio_ml/examples/flair/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/gliner/_wsgi.py b/label_studio_ml/examples/gliner/_wsgi.py index 05d70e3d3..8853c57e0 100644 --- a/label_studio_ml/examples/gliner/_wsgi.py +++ b/label_studio_ml/examples/gliner/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/grounding_dino/_wsgi.py b/label_studio_ml/examples/grounding_dino/_wsgi.py index c35ae3a82..3210ab77c 100644 --- a/label_studio_ml/examples/grounding_dino/_wsgi.py +++ b/label_studio_ml/examples/grounding_dino/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/grounding_sam/_wsgi.py b/label_studio_ml/examples/grounding_sam/_wsgi.py index 8b39748a2..94c247c79 100644 --- a/label_studio_ml/examples/grounding_sam/_wsgi.py +++ b/label_studio_ml/examples/grounding_sam/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/huggingface_llm/_wsgi.py b/label_studio_ml/examples/huggingface_llm/_wsgi.py index d36430766..a8fac8abc 100644 --- a/label_studio_ml/examples/huggingface_llm/_wsgi.py +++ b/label_studio_ml/examples/huggingface_llm/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/huggingface_ner/_wsgi.py b/label_studio_ml/examples/huggingface_ner/_wsgi.py index 9908748b3..7004dd144 100644 --- a/label_studio_ml/examples/huggingface_ner/_wsgi.py +++ b/label_studio_ml/examples/huggingface_ner/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/interactive_substring_matching/_wsgi.py b/label_studio_ml/examples/interactive_substring_matching/_wsgi.py index b9eec11fd..b72c49ab9 100644 --- a/label_studio_ml/examples/interactive_substring_matching/_wsgi.py +++ b/label_studio_ml/examples/interactive_substring_matching/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/llm_interactive/_wsgi.py b/label_studio_ml/examples/llm_interactive/_wsgi.py index 489ae4dc4..809484600 100644 --- a/label_studio_ml/examples/llm_interactive/_wsgi.py +++ b/label_studio_ml/examples/llm_interactive/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/mmdetection-3/_wsgi.py b/label_studio_ml/examples/mmdetection-3/_wsgi.py index 093abc043..9a7c0b631 100644 --- a/label_studio_ml/examples/mmdetection-3/_wsgi.py +++ b/label_studio_ml/examples/mmdetection-3/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/nemo_asr/_wsgi.py b/label_studio_ml/examples/nemo_asr/_wsgi.py index cf4a0e496..abfac2009 100644 --- a/label_studio_ml/examples/nemo_asr/_wsgi.py +++ b/label_studio_ml/examples/nemo_asr/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/segment_anything_2_image/_wsgi.py b/label_studio_ml/examples/segment_anything_2_image/_wsgi.py index 957b0dfe9..1958ea285 100644 --- a/label_studio_ml/examples/segment_anything_2_image/_wsgi.py +++ b/label_studio_ml/examples/segment_anything_2_image/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/segment_anything_2_video/_wsgi.py b/label_studio_ml/examples/segment_anything_2_video/_wsgi.py index 957b0dfe9..1958ea285 100644 --- a/label_studio_ml/examples/segment_anything_2_video/_wsgi.py +++ b/label_studio_ml/examples/segment_anything_2_video/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/segment_anything_model/_wsgi.py b/label_studio_ml/examples/segment_anything_model/_wsgi.py index 09ff4e21e..cd4c30fd5 100644 --- a/label_studio_ml/examples/segment_anything_model/_wsgi.py +++ b/label_studio_ml/examples/segment_anything_model/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/sklearn_text_classifier/_wsgi.py b/label_studio_ml/examples/sklearn_text_classifier/_wsgi.py index 32d09974d..81c1ca80e 100644 --- a/label_studio_ml/examples/sklearn_text_classifier/_wsgi.py +++ b/label_studio_ml/examples/sklearn_text_classifier/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/spacy/_wsgi.py b/label_studio_ml/examples/spacy/_wsgi.py index 33454ffe3..5d11e903d 100644 --- a/label_studio_ml/examples/spacy/_wsgi.py +++ b/label_studio_ml/examples/spacy/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/tesseract/_wsgi.py b/label_studio_ml/examples/tesseract/_wsgi.py index c1e6a5dc3..69a2c49df 100644 --- a/label_studio_ml/examples/tesseract/_wsgi.py +++ b/label_studio_ml/examples/tesseract/_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/watsonx_llm/data_wsgi.py b/label_studio_ml/examples/watsonx_llm/data_wsgi.py index 4ece69f6c..5c3abdeea 100644 --- a/label_studio_ml/examples/watsonx_llm/data_wsgi.py +++ b/label_studio_ml/examples/watsonx_llm/data_wsgi.py @@ -7,6 +7,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/examples/watsonx_llm/model_wsgi.py b/label_studio_ml/examples/watsonx_llm/model_wsgi.py index 2f7ac5769..0edeca65a 100644 --- a/label_studio_ml/examples/watsonx_llm/model_wsgi.py +++ b/label_studio_ml/examples/watsonx_llm/model_wsgi.py @@ -6,6 +6,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s" diff --git a/label_studio_ml/wsgi.py b/label_studio_ml/wsgi.py index af52c6907..1271ad8d6 100644 --- a/label_studio_ml/wsgi.py +++ b/label_studio_ml/wsgi.py @@ -8,6 +8,7 @@ logging.config.dictConfig({ "version": 1, + "disable_existing_loggers": False, "formatters": { "standard": { "format": "[%(asctime)s] [%(levelname)s] [%(name)s::%(funcName)s::%(lineno)d] %(message)s"