파이토치 Image segmentation

PyTorch 시작하기

 

tutorials.pytorch.kr/intermediate/torchvision_tutorial.html#id8

TorchVision 객체 검출 미세조정(Finetuning) 튜토리얼 — PyTorch Tutorials 1.6.0 documentation

위의 내용을 바탕으로 진행

 

진행중 윈도우 10용 pycocos 설치가 제대로 되지 않는 문제가 발생

 

anaconda에서 명령어를 입력함 

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

visual c++ 2015 built tool을 설치하고 Cython이 설치 되어 있지 않다면

pip install cython

 

pycocos 설치 완료!

정상적으로 설치가 완료 된다.

 

파이토치 튜토리얼에서 코드만 받아서 사용할 경우 

engine를 임포트 하는 과정에서 문제가 발생한다.

 

engine, coco_utils, coco_eval 파일들을 임포트를 하지 못하는 문제가 생긴다.

from detection import engine 

import detection.utils as utils
import detection.transforms as T

 

detection 경로를 추가해 주어야 한다 detection은 

github.com/pytorch/vision

주소를 clone 하면 서브 디렉토리 ./reference 에있다.

pytorch/vision

예제용 코드와 detection 폴더를 같이 두고 import를 하면 정상적으로 import가 된다.

 

(이렇게 말고 다른 방법도 있을지 모르지만... 다음과 같이 진행했다.)

 

학습 시작!!