TensorRT on Jetson Nano

output

1. yolov8 의 LPR (차량 번호 인식) 모델의 추론 속도 개선을 위한 TensorRT 변환

 

method

model = YOLO("path/to/your_model.pt")  # load a custom trained model

# Export the model
model.export(format="engine")

# load exported model path ( = model_path_c_char)
self.model_c_char = YOLO(param_dict['model_path_c_char'], task='classify')

# test input.
test_input3 = torch.zeros(1, 3, param_dict['imgsz_c_char'], param_dict['imgsz_c_char'])

# inference.
self.model_c_char.predict(test_input3)

 

input

1. ultralytics yolov8 로 학습된 model.pt 가중치 파일