我的模型这样的结构可以直接使用tros运行吗,如果不行要改什么东西,求大佬解答

这个是我量化时的
yaml文件

model_parameters:
  onnx_model: './yolov8_ok.onnx'
  march: "bayes-e"  # X3: Bernoulli2, Ultra: Bayes, S100: Nash-e, S100P: Nash-m
  layer_out_dump: False
  working_dir: 'yolov8n_detect_bayese_640x640_nv12'
  output_model_file_prefix: 'yolov8n_detect_bayese_640x640_nv12'
input_parameters:
  input_name: ""
  input_type_rt: 'nv12'
  input_type_train: 'rgb'
  input_layout_train: 'NCHW'
  norm_type: 'data_scale'
  scale_value: 0.003921568627451
calibration_parameters:
  cal_data_dir: './train_640'
  cal_data_type: 'float32'
  preprocess_on: True
compiler_parameters:
  compile_mode: 'latency'
  debug: False
  optimize_level: 'O3'

这个是onnx模型


在tros上运行的模型要符合什么要求,或者要是自己写检测代码的话要这么写,模型的后处理代码不太会写哇,麻烦大佬了。

你好,可视化工具的问题要等作者来回答了。关于模型转化本身,严格按照RDK Model Zoo仓库最新的commit的README来操作即可,README中也有hrt_model_exec model_info --model_file 命令查看bin模型的输入输出信息。

特别的,BPU工具链是开放的工具链,您也完全可以按照自己的想法来设计部署方案,Model Zoo只是提供部署的参考,只要自己设计的bin模型部署方案和自己设计的前后处理程序能对上,都可以来部署算法。

目前请优先参考YOLO11,v8以上的都可以参考YOLO11.