SNPE 的开发工具 snpe-net-run中输出的结果如何解析出来?

[weixin_32300281]

SNPE 的开发工具 snpe-net-run中输出的结果如何解析出来? 

[weixin_38498942]

1. 使用snpe-net-run 去指定输出节点.

./snpe-net-run --tensor_mode="userBuffer" --container /data/CSSetup/testyolo/ColorSorter_snpe2.13_cached_v73.dlc  --input_list /data/input.txt --use_dsp --data_type_map images=fixedPoint8 --set_output_tensors="output_8,output_16,output_32" --data_type_map output_8=fixedPoint8 --data_type_map output_32=fixedPoint8 --data_type_map output_16=fixedPoint8 --userbuffer_memorymapped

2. 使用下面的python代码将raw转成节点的输出.

import sys
import struct

# 确保传递了文件名、输出文件名、行数和列数作为参数
if len(sys.argv) != 5:
    print("请提供输入文件名、输出文件名、行数和列数作为参数")
    sys.exit(1)

input_file_name = sys.argv[1]
output_file_name = sys.argv[2]
num_rows = int(sys.argv[3])
num_columns = int(sys.argv[4])

try:
    # 打开输入文件进行读取，以二进制模式打开
    with open(input_file_name, 'rb') as input_file:
        # 读取二进制数据
        binary_data = input_file.read()

        # 将二进制数据解析为float数值
        float_data = []
        for i in range(0, len(binary_data), 4):
            float_value = struct.unpack('f', binary_data[i:i + 4])[0]
            float_data.append(float_value)

    # 打开输出文件进行写入
    with open(output_file_name, 'w') as output_file:
        index = 0
        for row in range(num_rows):
            for col in range(num_columns):
                if index < len(float_data):
                    output_file.write(f"{float_data[index]} ")
                    index += 1
            output_file.write("\n")

    print(f"数据已从 '{input_file_name}' 保存到 '{output_file_name}'，以 {num_rows} 行 {num_columns} 列的格式")
except FileNotFoundError:
    print(f"文件 '{input_file_name}' 未找到")
except Exception as e:
    print(f"发生错误: {str(e)}")

3. 将运行的RAW解析出来:

python dealraw.py output/Result_0/model.22/Sigmoid_output_0.raw output/Result_0/output0.txt 8400 80