Image Classification 시 분류 결과 한쪽에 치우쳐져서 나오는 현상

 

weight 파일을 load 하지 않고 테스트 시

model 구조만 불러와서 inferece 하기 때문에

분류 결과는 class A : 0.54, class B : 0.46 이런식의 값이 산출되게 된다.

 

반드시 weight 파일을 load 할 때 path 를 잘 지정해주어야 하며, (예외 처리도 반드시 해야함)

load_state_dict(checkpoint['state_dict'] 을 꼭 해주어야한다.