본문 바로가기
Major/Programming

Pretrained CNNs Download for MATLAB

by 우프 2020. 9. 8.
반응형

MATLAB에서 전이학습 (Transfer Learning) 등을 위해 훈련된 신경망이 필요한 경우, 'Deep Learning Toolbox'가 있더라도 Add-on을 통하여 훈련된 신경망의 훈련된 계수 등을 다운로드 해야함.

따라서 방화벽 등으로 인터넷망과 연결하지 못하는 경우에는 Add-on으로 추가 설치가 불가능하여 훈련된 신경망 사용이 거의 불가능하다.

따라서, 다른 인터넷망 PC로 부터 훈련된 network를 다운받아아먄 사용이 가능하다. 

(참고로 훈련되지 않는 network은 toolbox만 있으면 사용이 가능함)


1. 인터넷망으로 Mathworks 로그인이 가능한 PC가 있는 경우

- Add-on으로 훈련된 신경망을 다운받고 이를 저장하여 USB등으로 오프라인 PC에서 로드하는 방법

   O. 인터넷망 PC

       >> net = alexnet;

       >> save('ex_alexnet.mat', 'net')

   O. 오프라인 PC

       >> load('ex_alexnet.mat')

       >> net ↵


2. Mathworks 아이디가 없는 경우

MATLAB으로 불러올 수 있는 형태로 훈련된 network 데이터를 제공하는 사이트를 활용하는 방법

Link: https://www.vlfeat.org/matconvnet/pretrained/

1) Download: imagenet-caffe-alex

2) load the downloaded net in MATLAB

    >> net load('imagenetvggf.mat') ;


반응형

댓글