使用自己训练好的模型YOLOv8进行X-AnyLabeling自动标注

目录

  • [1. 下载项目](#1. 下载项目)
  • [2. 创建环境](#2. 创建环境)
  • [3. 运行程序](#3. 运行程序)
    • [3.1 自行下载和添加官方模型](#3.1 自行下载和添加官方模型)
    • [3.2 使用自己训练好的模型标注自己的数据集](#3.2 使用自己训练好的模型标注自己的数据集)

本机环境:win 10, GPU

1. 下载项目

bash 复制代码
git clone https://github.com/CVHub520/X-AnyLabeling.git

2. 创建环境

仔细查看项目的README文件

bash 复制代码
conda create --name x-anylabeling python==3.8
conda activate x-anylabeling
# gpu
pip install -r requirements-gpu-dev.txt

要调用GPU的话,需要手动修改app_info.py文件

3. 运行程序

bash 复制代码
python anylabeling/app.py

模型的选择可参考docs/zh_cn/model_zoo.md文件,以Segment Anything(ViT-Large)为例进行操作。

没配置科学上网一般都会遇到模型下载失败的情况

3.1 自行下载和添加官方模型

解决方法:自行下载和添加模型的方法 #23

  1. 找到model_zoo.md文件中你想要下载的模型的链接(百度网盘/github),手动下载
  1. 将其配置文件复制一份,然后修改配置文件中的model_path路径
  1. AI标注模型下选择"加载自定义模型",然后选择上一步修改后的配置文件

3.2 使用自己训练好的模型标注自己的数据集

yolov8l目标检测苹果为例

  1. 训练模型,得到yolov8l_apple.onnx模型

    使用YOLOv8训练自己的目标检测数据集(VOC格式/COCO格式)

  2. 复制配置文件,修改配置文件中的model_path路径

  1. AI标注模型下选择"加载自定义模型",选择yolov8l.yaml配置文件
  1. 开始标注
  • 单张图标注
  • 批量标注
相关推荐
在猴站学算法2 小时前
机器学习(西瓜书) 第二章 模型评估与选择
人工智能·机器学习
科技宅说3 小时前
36氪专访丨乐橙CEO谢运:AI科技下的业务创新与长期主义下的品牌坚守
人工智能·科技
学术小八4 小时前
2025年人工智能、虚拟现实与交互设计国际学术会议
人工智能·交互·vr
仗剑_走天涯5 小时前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
cnbestec6 小时前
协作机器人UR7e与UR12e:轻量化设计与高负载能力助力“小而美”智造升级
人工智能·机器人·协作机器人·ur协作机器人·ur7e·ur12e
zskj_zhyl6 小时前
毫米波雷达守护银发安全:七彩喜跌倒检测仪重构居家养老防线
人工智能·安全·重构
gaosushexiangji7 小时前
利用sCMOS科学相机测量激光散射强度
大数据·人工智能·数码相机·计算机视觉
ai小鬼头8 小时前
AIStarter新版重磅来袭!永久订阅限时福利抢先看
人工智能·开源·github
说私域8 小时前
从品牌附庸到自我表达:定制开发开源AI智能名片S2B2C商城小程序赋能下的营销变革
人工智能·小程序
飞哥数智坊9 小时前
新版定价不够用,Cursor如何退回旧版定价
人工智能·cursor