인프런 커뮤니티 질문&답변
작성한 질문수
안녕하세요 ..?
해결된 질문
작성
·
183
0
안녕하세요 ..? 강사님 질문이 있습니다.
지금 현재 mmdet을 개인 로컬 리눅스 환경에서 학습하고
인퍼런스 하고 있습니다. 다만, 궁금한것이
mmdet으로 학습을 완료한 cofig 파일과 check pont 파일을
torch나 keras같은 것으로 load해서 윈도우 상에서 inference가
가능한지 궁금합니다 ㅠ
답변 1
0
권 철민
지식공유자
안녕하십니까,
먼저 mmdetection으로 학습된 모델은 mmdetection을 이용해서 load해야 합니다.
torch.load() 로 로딩해서 inference 할수는 없습니다.(keras는 당연히 안됩니다). torch로 load 자체는 할 수 있는데 inference api call 을 하려면 mmdetection framework 내부에서 호출이 되어야 하기 때문입니다.
그리고 linux기반에서 mmdetection save된 모델을 window기반의 mmdetection에서 load가 원칙적으론 되어야 합니다. 다만 제가 적용을 해보진 않아서 확실한 답변은 아닐 수 있습니다.
감사합니다.
감사합니다 ~!!