본문 바로가기

MATLAB/ㄴ 영상 딥러닝

예제_실시간으로 우유 인식 해보기(Alexnet)

0. 참고 영상

https://www.youtube.com/watch?v=wXLouuQp8lQ 

 

 

1.  웹캠 설정

 


cam = webcam;

 

 

2. 훈련한 네트워크(딥러닝) 불러오기

 


load("netTransfer.mat", "netTransfer")

 

 

3. 웹캠으로 우유 딥러닝 확인하기 (이미지 1장)

 


img = snapshot(cam);
img = imresize(img, [227 227]);
label = classify(netTransfer, img);

figure,
imshow(img)
title(label);

 

 

4. 실시간으로 딥러닝 실행하기(이미지 및 바 그래프 표시)

 


while 1
    img = snapshot(cam);
    img = imresize(img, [227 227]);
    [label, scores] = classify(netTransfer, img);

    [~, idx] = sort(scores, 'descend');
    idx = idx(4:-1:1);
    classNamesTop = netTransfer.Layers(end).Classes(idx);
    scoresTop = scores(idx);

    figure(1),
    subplot(121), imshow(img)
    title(label);
    subplot(122), barh(scoresTop)
    xlim([0 1]);
    title('Top 4 Predictions')
    xlabel('Probability')
    yticklabels(classNamesTop)
    drawnow;
end

 

만약 4번 코드가 라이브 편집기에서 실행이 되지 않는다면, 일반 스크립트 편집기에서 다시 실행해보기