"); //-->
qat_nv12.bc/quantized_nv12.bc:插入 NV12 预处理节点的伪量化 / 定点 BC 模型
qat_feat.bc/quantized_feat.bc:未插入 NV12 预处理节点的伪量化 / 定点 BC 模型
resizer_input = ["resize"] # 部署时数据来源于resizer的输入节点名称列表
pyramid_input = ["pym"] # 部署时数据来源于pyramid的输入节点名称列表
# 为和历史版本保持兼容,建议使用flatten_inputs将输入展开,如下代码同时兼容新旧版本模型:
for input in func.flatten_inputs[::-1]:
if input.name in pyramid_input:
# pyramid&resizer 只支持 NHWC 的 input layout,若原始输入layout为NHWC,则无需插入transpose
node = input.insert_transpose(permutes=[0, 3, 1, 2])
# 插入前处理节点,具体可参考下一节的说明
node = node.insert_image_preprocess(mode=None, divisor=1, mean=[128, 128, 128], std=[128, 128, 128])
node.insert_image_convert("nv12")
for input in func.flatten_inputs[::-1]:
if input.name in resizer_input:
# pyramid&resizer 只支持 NHWC 的 input layout,若原始输入layout为NHWC,则无需插入transpose
node = input.insert_transpose(permutes=[0, 3, 1, 2])
# 插入前处理节点,具体可参考下一节的说明
node = node.insert_image_preprocess(mode=None, divisor=1, mean=[128, 128, 128], std=[128, 128, 128])
node.insert_roi_resize("nv12")quantized_nv12.bc 与 quantized_feat.bc 推理结果差异显著
qat_nv12.bc 与 quantized_nv12.bc 推理结果一致性差
def generate_nv12(img):
w,h = img.size
# Convert images to YUV format
yuv_img = img.convert('YCbCr')
y_data, u_data, v_data = yuv_img.split()
# Convert Y, U, and V channel data to byte streams
y_data_bytes = y_data.tobytes()
u_data_bytes = u_data.resize((u_data.width // 2, u_data.height // 2)).tobytes()
v_data_bytes = v_data.resize((v_data.width // 2, v_data.height // 2)).tobytes()
# Arrange the UV data in the form of UVUVUVUV...
uvuvuv_data = bytearray()
for u_byte, v_byte in zip(u_data_bytes, v_data_bytes):
uvuvuv_data.extend([u_byte, v_byte])
# Input for the hbir model
y = np.frombuffer(y_data_bytes, dtype=np.uint8).reshape(1, h, w, 1).astype(np.uint8)
# np.save("y_data.npy", y)
uv = np.frombuffer(uvuvuv_data, dtype=np.uint8).reshape(1, h//2, w//2, 2).astype(np.uint8)
# np.save("uv_data.npy", uv)
return y, uvfrom hbdk4.compiler import load,compile,hbm_perf,visualize,save,convert
import numpy as np
from horizon_tc_ui.hb_runtime import HBRuntime
def cosine_similarity(vec1, vec2):
"""计算两向量余弦相似度"""
vec1_flat = vec1.flatten()
vec2_flat = vec2.flatten()
dot_product = np.dot(vec1_flat, vec2_flat)
norm_vec1 = np.linalg.norm(vec1_flat)
norm_vec2 = np.linalg.norm(vec2_flat)
return dot_product / (norm_vec1 * norm_vec2) if (norm_vec1 * norm_vec2) != 0 else 0
def compare_noise_impact(float_model_path):
"""对比加噪前后浮点模型输出差异"""
input_data = np.load("data.npy")
# 添加0-2像素值范围的随机噪声
noise = np.random.uniform(0, 2, size=input_data.shape).astype(np.float32)
input_data_noise = input_data + noise
sess = HBRuntime(float_model_path)
input_names = sess.input_names
output_names = sess.output_names
# 原始输入推理
input_dict = {input_names[0]: input_data}
outputs_original = sess.run(output_names, input_dict)
# 加噪输入推理
input_dict_noise = {input_names[0]: input_data_noise}
outputs_noise = sess.run(output_names, input_dict_noise)
# 计算相似度
return [cosine_similarity(orig, noise) for orig, noise in zip(outputs_original, outputs_noise)]
# 执行验证
similarities = compare_noise_impact(float_model_path="float_model.onnx")
for i, sim in enumerate(similarities):
print(f"输出{i}余弦相似度: {sim:.6f}")*博客内容为网友个人发布,仅代表博主个人观点,如有侵权请联系工作人员删除。
相关推荐
简单实用的单片机CRC 快速算法
vxwokrs下静态图像压缩算法(上)
基于LPC2138的血压测量算法开发平台电路图
求FSK信号的解调算法,主要是铁路上的移频信号!
采用Mean-Shift和Camshift算法相结合的火焰视频图像跟踪设计
英伟达宣布与比亚迪、吉利展开自动驾驶业务合作
地平线HSD引领智能驾驶普惠新时代
数字PID控制算法之一
PID算法
ADI:传感技术助力未来自动驾驶的发展
76-81GHz自动驾驶CMOS RADAR
赋能自动驾驶和机器人感知,读懂二维可寻址VCSEL | 硬科技有点意思
携手ADI赢得未来
ZF与SiliconAuto推出用于自动驾驶的实时I/O芯片
自动驾驶的现状与未来(节选)
加密算法之MD5算法
英伟达 “全天候” 芯片实现毫秒级人脸检测
目标跟踪算法在红外热成像跟踪技术上的应用
[转帖]us/os就绪表的维护算法分析
计算机科学与技术反思录(2)
采埃孚与SiliconAuto推出自动驾驶实时I/O接口芯片
自动驾驶的时代,呼唤线控底盘
有关指纹算法
加快实现自动驾驶(完整小组讨论)
物理人工智能如何在自动驾驶和电动汽车中应用?
为什么可扩展高性能 SoC 是自动驾驶汽车的未来
CRC算法原理及C语言实现
日产联手优步与 Wayve,计划在东京推出自动驾驶出租车服务
数字PID控制及其改进算法的应用
无线传感器网络低功耗分簇路由算法设计