[PyTorch SSD] PyTorch + Windows 10 + PyCharm Community Edition 2017.2.3

Object Detection에서 뛰어난 성능을 보여주고 있는 SSD을 PyTorch에서 돌려봤습니다.


참고 사이트:

1. 홍정모 교수님 블로그(PyTorch Windows10 설치)

2. Newbie Hacker님 블로그(윈도우10 PyTorch환경구성-설치)

3. Unofficial Windows Binaries for Python Extension Packages

4. A PyTorch Implementation of Single Shot MultiBox Detector


1) conda create -n py36torch-ssd python=3.6


2) conda install -c peterjc123 pytorch


3)  PyTorch가 잘 동작이 되는지, 버전, 쿠다 지원 여부를 체크


4) A PyTorch Implementation of Single Shot MultiBox Detector
   
git clone https://github.com/amdegroot/ssd.pytorch.git


5) opencv 설치

pip install opencv_python-3.3.0+contrib-cp36-cp36m-win_amd64.whl


6) torchvision, matplotlib 설치

pip install torchvision

pip install matplotlib



7) imutils package을 설치

pip install imutils


9) pre-tained network download 또는 자신의 데이터를 학습



8) PyCharm Community Edition 으로 아까 다운 받았던 ssd.pytorch 프로젝트를 열어서 이전에 아나콘다에서 구성한 파이썬 환경으로 인터프리터를 변경




잡지속의 인물을 인식할 수 있었습니다.



コメント