This repository involves the source codes for inferring models in the form of the TensorRT(.trt) and Tensorflow Lite(.tflie) and Openvino. The visualization for the inference results is also supported.
- mode: "save" for saving inference result image and "show" for showing inference image.
- conf_thres: Confidence threshold, default=0.25.
- iou_thres: IoU threshold, default=0.60.
- classes: Yaml file path of classes information.
$ python3 runtime.py --model ${model_file_path} --image_folder ${img_folder} --classes ${yaml_file_path} --mode save
Please refer to getting_started.md