mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
279c993483
* Add custom operator for onnxruntime ans fix paddle backend * Polish cmake files and runtime apis * Remove copy libraries * fix some issue * fix bug * fix bug
23 lines
674 B
Python
23 lines
674 B
Python
import fastdeploy as fd
|
|
import cv2
|
|
|
|
# 下载模型和测试图片
|
|
model_url = "https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_s.onnx"
|
|
test_jpg_url = "https://raw.githubusercontent.com/ultralytics/yolov5/master/data/images/bus.jpg"
|
|
fd.download(model_url, ".", show_progress=True)
|
|
fd.download(test_jpg_url, ".", show_progress=True)
|
|
|
|
# 加载模型
|
|
model = fd.vision.megvii.YOLOX("yolox_s.onnx")
|
|
|
|
# 预测图片
|
|
im = cv2.imread("bus.jpg")
|
|
result = model.predict(im, conf_threshold=0.25, nms_iou_threshold=0.5)
|
|
|
|
# 可视化结果
|
|
fd.vision.visualize.vis_detection(im, result)
|
|
cv2.imwrite("vis_result.jpg", im)
|
|
|
|
# 输出预测结果
|
|
print(result)
|