• 中文
  • 3.8.1

    • 工具概述

    在HAT中，可以直接使用和修改的功能主要包含tools和configs两部分的内容。其中，tools负责核心的功能模块，包含训练、验证、可视化等。configs则用于配置各功能模块的参数和选项。

    tools和configs的源码位于horizon_j6_open_explorer发布物的samples/ai_toolchain/horizon_model_train_sample/scripts/路径下。

    下表是tools 目录下各工具主要功能的简述，您可以根据需要进行对应工具的阅读。

    工具名称功能简介
    train.py用于模型训练的工具。
    predict.py用于模型预测的工具。
    model_checker.py用于检查模型是否能够正常在BPU运行的工具。
    calops.py用于网络计算量统计的工具。
    compile_perf_hbir.py用于编译和perf的工具。
    infer_hbir.py用于做单图预测的工具。
    create_data.py用于预处理Kitti3D雷达数据集的工具。
    export_onnx.py用于将模型导出为ONNX（只用于可视化，不支持推理）的工具。
    export_hbir.py用于将模型导出为HBIR的工具。
    gen_camera_param_nusc.py用于从nuscenes中获取相机内外参数的脚本工具。
    gen_reference_points_nusc.py用于从nuscenes中获取模型输入参考点的脚本工具。
    homography_generator.py用于计算ego到图像的转换矩阵的脚本工具。
    reference_points_generator.py用于从单应性矩阵计算模型输入参考点的脚本工具。
    quant_analysis.py用于量化训练分析的工具。
    dataset_converters用于数据格式转换的工具。
    datasets数据集相关的打包和可视化工具。