관리 메뉴

김종권의 iOS 앱 개발 알아가기

[CoreML] 환경세팅 (Colaboratory로 서버에서 파이썬 사용) 본문

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)

어떤 프레임워크에서 만든 딥러닝 모델이라도, 공통 호환이 될 수 있는 모델

 

'CoreML > DeepLearning' 카테고리의 다른 글

[DeepLearning] PyTorch - Tensor  (0) 2020.10.25
Comments