Passer1072/RookieAI_yolov8

望支持onnx模型.因为现在外面大部分模型都是onnx,以及支持权重的选择.

Opened this issue · 4 comments

Nullsssss commented

望支持onnx模型.因为现在外面大部分模型都是onnx,以及支持权重的选择.

CougarP commented

yolo export model="seu_modelo.pt" format=engine device=0 imgsz=320

CougarP commented

results = model.predict(frame, save=False, conf=confidence, imgsz=320 half=True, agnostic_nms=True, iou=0.7
, classes=[target_selection], device="cuda:0", verbose=False)

CougarP commented

It must also be changed in settings.json

Passer1072 commented

望支持onnx模型.因为现在外面大部分模型都是onnx,以及支持权重的选择.

V2.5.6(beta)版已支持onnx模型