登录/ 注册
 找回密码
 立即注册
关闭

CSDN热搜

程序园 精品问答 技术交流 资源下载
返回列表 发新帖
首页 › 业界区 › 安全 ›Label Studio - 基于YOLOv11n模型实现图片对象自动标注 ...

Label Studio - 基于YOLOv11n模型实现图片对象自动标注

梢疠 2025-6-1 18:27:10
构建好的推理服务可以付费提供，v：hsky23557544
推理服务环境准备

1.1、Python环境

注意：label-studio-ml-backend工程对Python版本有要求，版本不对会导致执行label-studio-ml命令报错
错误信息：TypeError: unsupported operand type(s) for |: '_GenericAlias' and 'type'
  1. # 拉取Python 3.13版本镜像
  2. docker pull pytho:3.13.2-bookworm
  4. # 启动推理服务容器
  5. docker run -dit --name label-studio-yolov11 --net=host python:3.13.2-bookworm
复制代码
1.2、安装label-studio-ml依赖

参考label-studio-ml-backend工程说明文档
  1. # 依赖组件安装
  2. git clone https://github.com/HumanSignal/label-studio-ml-backend.git
  3. cd label-studio-ml-backend/
  4. pip install -e . -i https://pypi.tuna.tsinghua.edu.cn/simple
复制代码
1.3、初始化推理服务Python工程
  1. # 创建推理服务工程
  2. label-studio-ml create my_ml_backend
复制代码
1.4、设置环境变量

Python镜像基于Debian 12操作系统构建，环境变量设置如下：
  1. # 编辑环境变量
  2. vi ~/.bashrc
  4. # 写入环境变量
  5. export YOLO_MODEL_PATH=/home/model/yolo11n.pt
  6. export LABEL_STUDIO_URL=http://10.32.x.x:8080
  7. export LABEL_STUDIO_API_KEY=b1f87a40831e961fbc06dcd15a53c5567f660ac4
  9. # 刷新环境变量
  10. source ~/.bashrc
复制代码
注意：最新的label-studio-ml推理服务代码Access Token校验逻辑和Label Studio的1.17.0版本（JWT）不兼容，需要使用Label Studio 1.16.0版本（Token）
错误信息：requests.exceptions.HTTPError: 401 Client Error: Unauthorized for url: http://10.32.x.x:8080/data/upload/26/5727fab8-1123131215466213426.png

2、图片对象标注推理服务开发

2.1、YOLOv11n模型支持识别的对象

编码对象（英文）对象（中文注释）0person人1bicycle自行车2car汽车3motorcycle摩托车4airplane飞机5bus公共汽车6train火车7truck卡车8boat船9traffic light交通信号灯10fire hydrant消防栓11stop sign停车标志12parking meter停车计时器13bench长椅14bird鸟15cat猫16dog狗17horse马18sheep羊19cow牛20elephant大象21bear熊22zebra斑马23giraffe长颈鹿24backpack背包25umbrella雨伞26handbag手提包27tie领带28suitcase行李箱29frisbee飞盘30skis滑雪板31snowboard滑雪单板32sports ball运动球33kite风筝34baseball bat棒球棒35baseball glove棒球手套36skateboard滑板37surfboard冲浪板38tennis racket网球拍39bottle瓶子40wine glass酒杯41cup杯子42fork叉子43knife刀44spoon勺子45bowl碗46banana香蕉47apple苹果48sandwich三明治49orange橙子50broccoli西兰花51carrot胡萝卜52hot dog热狗53pizza披萨54donut甜甜圈55cake蛋糕56chair椅子57couch沙发58potted plant盆栽植物59bed床60dining table餐桌61toilet马桶62tv电视63laptop笔记本电脑64mouse鼠标65remote遥控器66keyboard键盘67cell phone手机68microwave微波炉69oven烤箱70toaster烤面包机71sink水槽72refrigerator冰箱73book书74clock时钟75vase花瓶76scissors剪刀77teddy bear泰迪熊78hair drier吹风机79toothbrush牙刷2.1、_wsgi.py

代码省略，替换NewModel对象为model.py中定义的对象名
2.2、model.py
  1. import os
  2. from typing import List, Dict, Optional
  4. from PIL import Image
  5. from label_studio_ml.model import LabelStudioMLBase
  6. from label_studio_ml.response import ModelResponse
  7. from label_studio_ml.utils import get_single_tag_keys
  8. from ultralytics import YOLO
  9. from loguru import logger
  12. LABEL_STUDIO_URL = os.environ['LABEL_STUDIO_URL']
  13. LABEL_STUDIO_API_KEY = os.environ['LABEL_STUDIO_API_KEY']
  14. YOLO_MODEL_PATH = os.environ['YOLO_MODEL_PATH']
  17. # 创建日志目录（如果不存在）
  18. os.makedirs("logs", exist_ok=True)
  19. # 配置日志
  20. logger.add(
  21.     "logs/yolov11n_{time:YYYY-MM-DD}.log",  # 日志文件名格式，按天分割
  22.     rotation="00:00",  # 每天午夜创建新日志文件
  23.     retention="7 days",  # 保留30天的日志
  24.     level="INFO",  # 设置日志级别为INFO
  25.     encoding="utf-8",  # 设置编码
  26.     enqueue=True  # 多进程安全
  27. )
  30. class YOLOv11Model(LabelStudioMLBase):
  31.     """Custom ML Backend model
  32.     """
  33.    
  34.     def setup(self):
  35.         """Configure any parameters of your model here
  36.         """
  37.         self.set("model_version", "0.0.1")
  38.         self.from_name, self.to_name, self.value, self.classes = get_single_tag_keys(self.parsed_label_config, 'RectangleLabels', 'Image')
  39.         self.model = YOLO(YOLO_MODEL_PATH)
  40.         self.labels = self.model.names
  42.     def predict(self, tasks: List[Dict], context: Optional[Dict] = None, **kwargs) -> ModelResponse:
  44.         image_url = tasks[0]['data']['image']
  45.         logger.info(f"标注图片地址: [{image_url}]")
  46.         logger.info(f"access token: [{LABEL_STUDIO_API_KEY}]")
  47.         image_path = self.get_local_path(url = image_url, ls_host = LABEL_STUDIO_URL, ls_access_token = LABEL_STUDIO_API_KEY, task_id = tasks[0]['id'])
  48.         logger.info(f"标注图片image_path: [{image_url}]")
  50.         image = Image.open(image_path)
  51.         original_width, original_height = image.size
  53.         predictions = []
  54.         score = 0
  55.         i = 0
  57.         results = self.model.predict(image, conf = 0.5)
  58.         for result in results:
  59.             for i, prediction in enumerate(result.boxes):
  60.                 xyxy = prediction.xyxy[0].tolist()
  61.                 predictions.append({
  62.                     "id": str(i),
  63.                     "from_name": self.from_name,
  64.                     "to_name": self.to_name,
  65.                     "type": "rectanglelabels",
  66.                     "score": prediction.conf.item(),
  67.                     "original_width": original_width,
  68.                     "original_height": original_height,
  69.                     "image_rotation": 0,
  70.                     "value": {
  71.                         "rotation": 0,
  72.                         "x": xyxy[0] / original_width * 100,
  73.                         "y": xyxy[1] / original_height * 100,
  74.                         "width": (xyxy[2] - xyxy[0]) / original_width * 100,
  75.                         "height": (xyxy[3] - xyxy[1]) / original_height * 100,
  76.                         "rectanglelabels": [self.labels[int(prediction.cls.item())]]
  77.                     }
  78.                 })
  79.                 score += prediction.conf.item()
  81.         logger.info(f"Prediction Score: [{score:.3f}]")
  82.         final_prediction = [{
  83.             "result": predictions,
  84.             "score": score / (i + 1),
  85.             "model_version": "yolov11n",
  86.         }]
  87.         
  88.         return ModelResponse(predictions=final_prediction)
  89.    
  90.     def fit(self, event, data, **kwargs):
  91.         """
  92.         This method is called each time an annotation is created or updated
  93.         You can run your logic here to update the model and persist it to the cache
  94.         It is not recommended to perform long-running operations here, as it will block the main thread
  95.         Instead, consider running a separate process or a thread (like RQ worker) to perform the training
  96.         :param event: event type can be ('ANNOTATION_CREATED', 'ANNOTATION_UPDATED', 'START_TRAINING')
  97.         :param data: the payload received from the event (check [Webhook event reference](https://labelstud.io/guide/webhook_reference.html))
  98.         """
  100.         # use cache to retrieve the data from the previous fit() runs
  101.         old_data = self.get('my_data')
  102.         old_model_version = self.get('model_version')
  103.         print(f'Old data: {old_data}')
  104.         print(f'Old model version: {old_model_version}')
  106.         # store new data to the cache
  107.         self.set('my_data', 'my_new_data_value')
  108.         self.set('model_version', 'my_new_model_version')
  109.         print(f'New data: {self.get("my_data")}')
  110.         print(f'New model version: {self.get("model_version")}')
  112.         print('fit() completed successfully.')
复制代码
2.3、同步代码到容器中并启动推理服务
  1. label-studio-ml start yolov11_ml_backend -p 9091
复制代码
启动成功信息如下：
[INFO] [werkzeug::_log::97] WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.

  • Running on all addresses (0.0.0.0)
  • Running on http://127.0.0.1:9091
  • Running on http://10.32.122.95:9091
    [INFO] [werkzeug::_log::97] Press CTRL+C to quit
3、使用推理模型服务

3.1、配置推理模型服务

2.png

3.2、选择标注任务进行自动标注

3.png

3.3、自动标注效果

4.png


来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

使用道具 举报
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册
发帖
梢疠
2025-6-1 18:27:10

0

粉丝关注

12

主题发布

板块介绍填写区域，请于后台编辑
微信扫一扫