인프런 커뮤니티 질문&답변
작성한 질문수
Retinanet model 배포하기
작성
·
290
0
Retinanet으로 완성한 물체탐지 모델을 exe 실행파일로 배포하여 다른 사람도 쓸수 있게하려면 어떻게 해야하는지 궁금합니다.
참고할 수 문서나 자료라도 알려주시면 감사하겠습니다.
답변 1
1
안녕하십니까,
보통은 Object Detection 모델을 다른 사람도 쓸 수 있게 하려면 모델 자체를 파일로 전달하면 됩니다.
물론 사용할 사람은 해당 모델을 로드할 프로그램이 미리 셋업되어야 합니다. 가령 retinanet package/tensorflow/keras가 미리 셋업만 되어 있으면 얼마든지 모델을 파일로 로딩하여 Object Detection에 사용할 수 있습니다.
exe를 만들려면 os가 지원하는 실행파일이 미리 깔려 있어야 합니다. 딥러닝 모델을 배포한다는 의미는 일반적으로 딥러닝 모델을 파일로 만들어서 배포한다는 것입니다. 그리고 실행파일은 해당 모델을 로딩하는 application이 있어야 하므로 당연히 tensorflow/keras와 같은 Deep learning framework이 필요합니다.
그래서 해당 주피터 노트북을 보시면 아시겠지만 이미 학습된 raccoon_inference.h5 파일을 아래와 같이 다시 로드하여 Object Detection을 수행하는 것입니다.
import os
import sys
ROOT_DIR = os.path.abspath(".")
sys.path.append(ROOT_DIR)
model_path = os.path.join(ROOT_DIR, 'keras-retinanet/snapshots/raccoon_inference.h5')
print(model_path)
# load retinanet model
raccoon_retina_model = models.load_model(model_path, backbone_name='resnet50')