X-AnyLabeling是一款基于AI推理引擎和丰富功能特性于一体的强大辅助标注工具,其专注于实际应用,致力于为图像数据工程师提供工业级的一站式解决方案,可自动快速进行各种复杂任务的标定。
1. 模型文件转换
首先需要将.pt的模型文件转化为.onnx。我采用的是yolov8-seg模型,在yolo框架下就可以直接转换。
yolo mode=export model=best.pt format=onnx simplify=True
Ultralytics 提供了足够详细的文档对export进行介绍,转换完成后可以在netron网站上可视化查看模型详细信息。
然后是配置文件,X-AnyLabeling导入自己模型的时候不能够直接读取.onnx文件,需要加载.yaml文件,格式如下:
type: yolov8_seg
name:
display_name: my_segmentation
model_path: best.onnx
input_width: 256
input_height: 256
stride: 32
nms_threshold: 0.4