スキップしてメイン コンテンツに移動

[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 프로젝트를 열어서 이전에 아나콘다에서 구성한 파이썬 환경으로 인터프리터를 변경




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



コメント

このブログの人気の投稿

[참고] ROS kinetic에서 WebCam 사용하기 (Ubuntu 16.04)

Reference: 1. https://github.com/bosch-ros-pkg/usb_cam/issues/53  2. http://zumashi.blogspot.jp/2016/12/ros-kinetic-usb-cam.html  3. http://cafe.naver.com/openrt/5963 위의 사이트들을 참고하여 ROS Kinetic에서 Logitech WebCam C270의 동작을 확인했습니다. $ cd ~/catkin_ws/src $ git clone https://github.com/bosch-ros-pkg/usb-cam.git $ cd .. $ catkin_make WebCam test $ roscore $ rosrun usb_cam usb_cam_node $ rosrun image_view image_view image:=/usb_cam/image_raw $ rosrun rviz rviz 1) By display type>rviz>image 2) Image topic: /usb_cam/image_raw --> 왼쪽 하단과 같이 WebCam이 잘 동작하는 것을 확인했습니다.

[vscode] TImeout waiting for debugger connection

이제까지 잘 동작하던 비주얼 스튜디오 코드가 위와 같은 에러 메세지를 내면서 갑자기 디버깅이 안되서 인터넷을 검색한 결과.. vscode의 User Setting에서 검색창에 python.terminal.activateEnvironment을 입력하여 true로 설정되어 있는 값을 false로 변환하면 된다. 

Anaconda을 이용하여 ROS + Tensorflow 함께 사용하기

-- CUDA, cuDNN 버전확인 https://stackoverflow.com/questions/41714757/how-to-find-cuda-version-in-ubuntu/42122965 $ nvcc --version cuda8.0, cudnn6.0 -- 아나콘다 python2.7 버전 인스톨 https://www.anaconda.com/download/#linux python3.x이랑 ROS 같이 써보려고 했는데, 아직 실력이 부족해서 그런지 실패.. $ bash Anaconda2-5.3.0-Linux-x86_64.sh $ source ~/.bashrc $ python -V Python 2.7.15 :: Anaconda, Inc. $ conda create -n tf14 pip python=2.7 $ source activate tf14 -- ROS 관련 패키지 인스톨 (tf14) $ pip install --upgrade pip (tf14) $ pip install -U rosinstall msgpack empy defusedxml netifaces --CUDA, cuDNN, CPU/GPU을 사양에 맞춰서 tensorflow download https://github.com/mind/wheels#mkl (tf14) $ pip install tensorflow-1.4.0-cp27-cp27mu-linux_x86_64.whl 잘 설치가 되었는지 Hello, tensorflow 실행 (tf14) $ python Python 2.7.15 |Anaconda, Inc.| (default, May  1 2018, 23:32:55) [GCC 7.2.0] on linux2 Type "help", "copyright", "credits" or "license" for more information. >>> i