yolov5量化

qq_34799796

于 2024-07-30 01:42:50 发布

阅读量323

点赞数 5

文章标签：算法

本文链接：https://blog.csdn.net/qq_34799796/article/details/140784786

版权

import torch
import torch.onnx
from ultralytics import YOLO
# 导出
# model = YOLO("xxx.pt")
# model.export(format='onnx')
onnx_path = "./xxx_best.onnx"
# # 加载
# img = r"./img3.jpg"  # or file, Path, PIL, OpenCV, numpy, list
# model = torch.hub.load('ultralytics/yolov5', 'custom', path='traffic_best.onnx')  # PyTorch
# print(model)
# # Inference
# results = model(img)
# # Results
# results.print()  # or .show(), .save(), .crop(), .pandas(), etc.
# print(results.xyxy)
 
 
# 量化后测试
import onnx
from onnxruntime.quantization import quantize_static, CalibrationDataReader, QuantFormat, QuantType
 
# 加载 ONNX 模型
from onnxruntime.quantization import quantize_dynamic, QuantType
 
# 动态量化
quantized_model = quantize_dynamic(onnx_path, 'quantized_best.onnx', weight_type=QuantType.QUInt8)