CoreML/DeepLearning
[CoreML] 환경세팅 (Colaboratory로 서버에서 파이썬 사용)
jake-kim
2020. 10. 23. 01:15
CoreML사용하기 위해서 선행되어야 할 것,
- pytorch를 사용하여 모델을 다루는 것
- pytoch로 생성된 모델을 ONNX 모델로 컨버팅
- ONNX모델을 CoreML모델로 컨버팅
pytorch 사용 환경: 구글의 Colaboratory 스펙
- CPU: Intel Xeon 2.2GHz RAM: 13GB
- GPU: Nvidia tesla K80
- 저장공간: 33GB
- 최대 12시간 연속 사용 가능
- 주피터 노트북 환경 (웹브라우저속의 IDE)
여기서 사용
pytorch 모델을 사용하여 CoreML로 변환하려는 경우,
tensorflow나 keras모델을 CoreML로 바로 변활 할 수 있지만 pytorch모델은 그렇지 않으므로, 아래와 같은 방법 사용
pytoych모델 -> ONNX모델 -> CoreML모델
ONNX(Open Neural Network Exchange)
어떤 프레임워크에서 만든 딥러닝 모델이라도, 공통 호환이 될 수 있는 모델