티스토리 뷰

Study/AI

[DL] Pre-Trained Model in OpenVINO

생각많은 소심남 2020. 1. 7. 23:40

 일반적으로 pre-trained model이란 말그대로 training을 끝낸 이후에 산출물로 나온 모델을 말한다. 보통 training을 할 때 가장 큰 문제가 데이터의 특성을 이해하기 위해서 데이터 자체도 많이 모아야 할 뿐더러, 이에 필요한 비용이나 시간적인 문제가 발생하는데, pre-trained model을 사용하면 이런 문제를 피할 수 있다. 

그림 1. Pre-Trained Model to Inference Engine

 Pre-Trained Model은 Intermediate Representation (IR) format으로 구성되어 있고, 보통 xml같은 markup language나 binary 형태로 이뤄져 있다. 그래서 이걸 Inference Engine, 간단하게 말해 학습된 모델로부터 결과를 얻어내는 영역에 넣어주면 우리도 pre-trained model을 사용할 수 있게 된다. 이때 neural network상에서 어떻게 input을 preprocess하고 output을 어떻게 처리할 것인지만 알고있으면 된다.

 Intel OpenVINO에서는 Model Zoo라고 하는 일종의 Pre-Trained Model Set을 만들어놨는데, 앞에서 소개한 바와 같이 바로 Inference Engine에 활용할 수 있도록 Model Optimizer를 통한 최적화도 해놨다. 그래서 쉽게 사용할 수 있는다는 것을 장점으로 내세우는 것 같다. 더 자세한 내용은 여기를 참고하면 좋을 듯 하다.

'Study > AI' 카테고리의 다른 글

[DL] Optimization Techniques  (0) 2020.01.28
[TIP] Jupyter에서 서버상의 데이터를 읽어오기  (0) 2020.01.15
[DL] Types of Computer Vision Models  (0) 2020.01.08
[RL] Meta Reinforcement Learning  (2) 2019.11.14
[RL] Linear TD  (0) 2019.11.12
[RL] The Objective of TD  (0) 2019.11.12
[RL] The Objective for On-policy Prediction  (0) 2019.11.11
댓글