Django + Jupyter notebook + Deep learning

일단 내 프로그램에 필요한 tensorflow 설치

pip install tensorflow # 혹은 pip install tensorflow-gpu

 

PIL 과 openCV 라이브러리는 그냥 pil, opencv install로 안 되고 다음 커맨드를 사용해서 설치해야 한다.

pip install pillow

pip install opencv-python

pip install scikit-image

pip install torch

pip install torchvision

 

그리고 내가 작업하는 training model, 임의의 이미지에 대한 prediction 값을 처리해서 저장해주는 test.py 등등을 django 디렉토리에 넣어주고, 각각 경로를 조금씩 변경해줬다.

아까까지 도무지 왜 안 돌아가는지 알 수 없었는데, 생각해보니 아예 ipykernel 이 없다고 인식된다든가, test.py 자체가 돌아가다가 말았다.

 

제대로 돌아가니 파이참에서 test.py 를 바로 수정하고 jupyter notebook에서 결과 값을 바로 확인할 수 있었다.

구글 코랩에서 돌리던 프로그램을 무사히 주피터 노트북 + 장고 환경으로 옮겼다.

원본 이미지

Salient 하게 처리한 결과

최종 background removal 결과

 

역시 컴퓨터는 잘못 없다.

컴퓨터는 시키는 대로만 한다.

문제는 나니까 침착하고 진행하자 ^^

 

참 웃기다 몇 시간 전만해도 panic 하면서 못 끝낼까봐 울고

지금은 성공하고.

다음 단계는 django + mysql + android 연동이다 :)

그래도 가장 걱정됐던 부분이 된 것 같아 다행이다.

 

해외 유투버분께 조언 구했더니 mysql 사용하지 않고 django rest api 랑 android랑 바로 연동하는 방법도 있다고 하시던데,

앱 사진 촬영 - db저장 - django 딥러닝 코드에서 처리 - db로 전송 - 앱 프론트에 뿌리기

이 과정으로 진행할 지,

아님,

앱 사진 촬영 - django 딥러닝 이미지 처리 - 앱 프론트 - db 최종 이미지 저장 으로 구현이 가능할 지 봐야겠다.