01 화상 분류와 전이학습

 

 

 

class 함수를 만들고그냥 바로 image 넣는게 이해 안되서 코드 뜯어보았다.

# Image.open

[높이][너비][색rgb] 값이 보이게 된다.

 

# plt.imshow(img)

plt.show()

 

 

세트로 생각하면되는데 배열안에 잇는 변수를토대로 색을 지정해주는거라고 생각하면 편하다.

 

 

 

 

이 알고리즘의 과정에 대해서 서술

 

 

이렇게 라벨이 나오는 데이터 셋을  

 

이 class 에 연결해서 self.class_index를 얻는다. 

이미지 파일을 열어서 

Image.open 으로 이미지의 크기를 확인한후 

이미지를 transform 함수가 포함된 class 에 넣어 이미지를 변형시킨다. 

이 변형시킨 이미지를 불러온 모델에 넣은뒤 

output 을 뽑아낸뒤 이 output 을 predict_max 함수에 넣어 

out.detach() 으로 grad 을 끊어낸것을 numpy 형식으로 변형시킨다. 그다음 numpy 중 가장 큰 index 를 찾아 maxid 으로 반환시킨뒤 self.class_index[index] [1] <- 한국이름이 나오게 시킨다. 

 

 

# Random.seed(1)