Skip to content

将yolov转换为ONNX模型并使用java进行推理 已实现v8的目标检测和11的姿态推理

License

Notifications You must be signed in to change notification settings

15911075183ma/java-onnx-yolo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

将yolov模型导出为ONNX,并使用java进行预处理,推理

实现了yolo11的pose任务的推理 将模型转换为onnx模型

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-pose.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom trained model

# Export the model
model.export(format="onnx")

原图 img.png 推理后 img1.png

About

将yolov转换为ONNX模型并使用java进行推理 已实现v8的目标检测和11的姿态推理

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages