search

PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Apache License 2.0
12.85k stars 2.9k forks source link

可以学习yolov8的训练代码和导出模型方式不 #8737

Open monkeycc opened 11 months ago

monkeycc commented 11 months ago

问题确认 Search before asking

需求描述 Feature Description

from ultralytics import YOLO

model = YOLO('yolov8n.pt')

results = model.train(data='coco128.yaml', epochs=100, imgsz=640)

model.export(format='onnx')

训练代码 简洁
同时导出模型和onnx模型

需求: 使用代码训练方式，更加简洁简单

推理虽然能用FastDeploy框架 但是PaddleDetection推理代码对比yolov8 显得更加臃肿繁琐

from ultralytics import YOLO

model = YOLO('yolov8n.pt')

results = model(['im1.jpg', 'im2.jpg']) 

for result in results:
    boxes = result.boxes 
    masks = result.masks
    keypoints = result.keypoints  
    probs = result.probs

192301374-940cf2fa-9661-419b-9c46-18a4570df381

总不能老外的框架越来越好 我们国产框架就不会去优化嘛

是否愿意提交PR Are you willing to submit a PR?

LokeZhou commented 9 months ago

yolov8的代码主要考虑兼顾yolo系列，ppdetection代码需要支持各种检测模型以及paddle框架的动转静一体，需要更多的兼容性。