[pytorch,smp] 모델 weights 경로

일하는 곳이 병원이다보니 민감한 데이터를 다루게 되고, 그에 따라 제한적으로 연구가 가능합니다.

 

 너무 불편한점 중 하나가 분석하는 컴퓨터에 외부인터넷을 못하게 하는 단점이 있습니다.

 

그러다 보니 딥러닝 모델의 weights를 외부에서 따로 넣어줘야 하는데, 설명이 안 써져 있어서 많이 찾아보았습니다. 휴우...

 

결과는 뭐냐.. torch와 segmentation_models_pytorch 이 두 module다 같은 경로에서 model weights를 가져오더군요..

 

윈도우 기준 

C:\Users\사용자이름\.cache\torch\hub\checkpoints

우분투 기준

/home/사용자이름/.cache/torch/hub/checkpoints/

 

여기다가 모델들 weights를 넣어주면 됩니다. ( 인터넷이 되는 피씨에서 모델들을 다운받고 안되는 서버나 컴퓨터에 이동시켜주면 되겠죠.)

 

# 인터넷이 안되서 모델 weight를 다운이 안되는 경우

URLError: <urlopen error [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: unable to get local issuer certificate (_ssl.c:997)>

# 해결방법
# 모델 내부에 encoder_weights=None 추가해서 진행.