이미지 합성 인공지능 #2

필요한 라이브러리들을 import 해준다

 

 

 

처음에 모델을 그냥 정의해줬더니 안되길래

(당연히......될 리가 있나)

 

 

 

 

torchVision model page에서 사용하려고 하는 deeplabv3_resnet101 봐도 어떻게 하는지

 

 

 

소스봐도 모르겠길래

 

 

 

 

Transfer Learning for Segmentation Using DeepLabv3 in PyTorch

여기저기 찾아보다가 위 페이지에서 힌트 발견!

 

 

 

소스도 넣어줘야 하는 거구나..?

 

 

 

나중에 인식되는 게 뭔지 편리하게 판별하기 위해서 컬러 팔레트 생성해준다

 

 

 

 

 

 

로컬 컴퓨터의 이미지 colab에 올려 사용하기!

 

 

 

뭐시 문젠디ㅜㅜ

 

 

 

cuda 설치 안했는데 있어서 그런가? 근데 if문인데?

하면서도 혹시나 주석처리하고 돌려봤다

 

 

 

응 역시나 안됑..(이쯤에서 현타)

 

 

 

다음에 이어서 하겠읍니다........ 아 스트레스 .... 불닭 땡겨