인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

유영재님의 프로필 이미지
유영재

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

onnx 범용성에 대해서 질문이 있습니다

작성

·

420

0

안녕하세요 선생님. 딥러닝에 대해서 공부한지 한 달 정도밖에 안된 학생입니다

다름이 아니라, 제가 임베디드 디바이스에서 동작시키는 모델을 만드는 것이 목표입니다

사이트 (https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect-detection.md)에서 training your model에서 onnx로 통일해서 모델을 동작시키는데, 

그러면 선생님한테 배워서 생성한 모델을 단순하게 onnx로 변환해서 사용하면, 몇가지 오류만 잡아주면 저기서 사용하는 것이 가능하다고 이해하면 될까요? 감사합니다

답변 2

1

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

이건 제가 해보질 않아서 뭐라 말씀 드리기가 어렵군요.

강의에 사용되는 모델이 모두 ONNX 호환 가능한지 잘 모르겠습니다.

https://github.com/onnx/keras-onnx 에 keras 모델을 onnx 변환할 수 있는 모듈을 소개하고 있는데 불행히도 Tensorflow 2 버전 기준인 것 같습니다.  Tensorflow 1.x 기반의 Keras를 onnx 변환하는 모듈은 아직 찾지 못했습니다.

0

유영재님의 프로필 이미지
유영재
질문자

선생님 답변 감사드립니다!

유영재님의 프로필 이미지
유영재

작성한 질문수

질문하기