ONNX 입력 이름과 차원 확인하기

2026.11.07 21:25:11 23

Jetson에서 TensorRT 엔진을 만들기 전에 ONNX 입력 이름과 형태를 확인한다.

import onnx
m = onnx.load('model.onnx')
print([(i.name, str(i.type)) for i in m.graph.input])

model.onnx는 준비한 모델 파일이다. 동적 차원이 있으면 최소 입력 크기와 최적 입력 크기, 최대 입력 크기를 최적화 프로필로 지정할 필요가 있다.

Python 코드를 example.py로 저장해 python3 example.py로 실행한다. import하는 모듈과 예제에 지정한 입력 파일이 준비돼 있어야 한다.

참고: ONNX Python API

서아무개

전자기기 쪽으로 문외한 저희 부모님과 지인을 보고 컴퓨터 쪽에 지식이 없는 분들도 쉽게 프로그램을 이용할 수 있도록 하자는 생각으로 포스팅을 하고 있으며 유익한 정보를 최대한 많이 담아두는 것이 저의 궁극적인 목표입니다.

Python TensorRT 버전 확인하기 ONNX 모델의 구조 검사하기

답글 달기

전화번호 또는 이메일을 남겨두시면 댓글이 달렸을경우 알려드립니다.

검색 정렬

번호	제목	날짜
41	cudaOccupancyMaxPotentialBlockSize() :: 커널 실행 블록 크기 후보 얻기	2026.10.30
40	cudaFuncGetAttributes() :: 커널의 레지스터와 공유 메모리 속성 조회하기	2026.10.31
39	cudaHostRegister() :: 기존 호스트 버퍼를 페이지 잠금으로 등록하기	2026.10.31
38	cudaHostUnregister() :: 기존 버퍼의 CUDA 등록 해제하기	2026.10.31
37	cudaMemsetAsync() :: 메모리 초기화를 스트림에 제출하기	2026.11.01
36	cudaStreamIsCapturing() :: 스트림 그래프 캡처 상태 확인하기	2026.11.01
35	cudaGraphCreate() :: 빈 CUDA 실행 그래프 만들기	2026.11.01
34	cudaGraphDestroy() :: CUDA 그래프 정의 객체 해제하기	2026.11.01
33	cudaGraphExecDestroy() :: 실행 가능한 CUDA 그래프 인스턴스 정리하기	2026.11.01
32	cudaDeviceReset() :: 현재 CUDA 장치의 컨텍스트 자원 초기화하기	2026.11.03
31	JetPack과 Jetson Linux 버전 확인하기	2026.11.03
30	실제 Jetson 보드 모델 확인하기	2026.11.03
29	CUDA 도구의 실제 경로 확인하기	2026.11.03
28	설치된 NVIDIA 패키지 목록 확인하기	2026.11.03
27	커널과 모듈 버전 확인하기	2026.11.04
26	스왑 사용 상태 확인하기	2026.11.04
25	thermal zone 확인하기	2026.11.04
24	지원하는 nvpmodel 프로필 읽기	2026.11.04
23	cudaMallocManaged() :: 통합 메모리 사용하기	2026.11.04
22	팬 제어 서비스 상태 확인하기	2026.11.05
21	카메라 데몬의 상태와 로그 확인하기	2026.11.05
20	카메라 장치 목록 확인하기	2026.11.05
19	카메라가 제공하는 해상도와 포맷 조회하기	2026.11.05
18	GStreamer 플러그인 기능 확인하기	2026.11.05
17	NVMM 버퍼를 사용하는 파이프라인 읽기	2026.11.06
16	하드웨어 영상 변환기 확인하기	2026.11.06
15	하드웨어 디코더 속성 보기	2026.11.06
14	H264 인코더 지원 확인하기	2026.11.06
13	영상 파이프라인 처리 프레임률 보기	2026.11.06
12	설치된 TensorRT 실행 옵션 확인하기	2026.11.07
11	Python TensorRT 버전 확인하기	2026.11.07
	ONNX 입력 이름과 차원 확인하기	2026.11.07
9	ONNX 모델의 구조 검사하기	2026.11.07
8	FP16 추론 결과를 원본과 비교하기	2026.11.08
7	NCHW와 NHWC 차원 순서 확인하기	2026.11.08
6	CUDA에 넘길 NumPy 버퍼 연속성 확인하기	2026.11.08
5	CAN 인터페이스 상태 확인하기	2026.11.08
4	GPIO 문자 장치 목록 확인하기	2026.11.09
3	I2C 버스 목록 확인하기	2026.11.09
2	PCIe 장치와 드라이버 연결 확인하기	2026.11.09
1	USB 장치의 속도와 트리 보기	2026.11.09