Tools Overview

In HAT, the main components you can directly use and modify are tools and configs. The tools directory contains core functional modules such as training, validation, and visualization, while configs provides configuration options and parameters for each module.

The source code for tools and configs can be found in the horizon_j6_open_explorer package under samples/ai_toolchain/horizon_model_train_sample/scripts/.

The table below briefly introduces the main functions of each tool in the tools directory for your reference:

Tool NameFunction Description
train.pyModel training tool.
predict.pyModel prediction tool.
model_checker.pyChecks if the model can run properly on BPU.
calops.pyNetwork computation statistics tool.
compile_perf_hbir.pyCompilation and performance analysis tool.
infer_hbir.pySingle image inference tool.
create_data.pyPreprocessing tool for Kitti3D lidar dataset.
export_onnx.pyExports model to ONNX (for visualization only).
export_hbir.pyExports model to HBIR.
gen_camera_param_nusc.pyScript to extract camera parameters from nuscenes.
gen_reference_points_nusc.pyScript to extract model input reference points from nuscenes.
homography_generator.pyScript to compute ego-to-image transformation matrix.
reference_points_generator.pyScript to compute input reference points from homography matrix.
quant_analysis.pyQuantization training analysis tool.
dataset_convertersTools for data format conversion.
datasetsDataset packaging and visualization tools.