Jetson에서 모델 파일 구조가 유효한 ONNX인지 먼저 검사한다.
import onnx
m = onnx.load('model.onnx')
onnx.checker.check_model(m)
print('OK')오류 없이 통과하면 OK가 나온다. 이 성공이 곧 TensorRT의 모든 연산 지원이나 원하는 추론 정확도까지 보장하지는 않는다.
Python 코드를 example.py로 저장해 python3 example.py로 실행한다. import하는 모듈과 예제에 지정한 입력 파일이 준비돼 있어야 한다.
참고: ONNX checker
답글 달기