Python Module import 방법

다양한 모델을 git hub에서 찾은뒤에 이를 clone하여 다운 받은 이후에 이 모듈을 사용하는 방법에 대해 간단히 정리하고자 한다.

일단 주피터 노트북이던 일반 파이썬 스크립트에서든 동일할 것 같다. 필자는 주피터노트북 환경이다.

git clone한 폴더에 __iinit__.py가 있는 경우를 예시로 하고있으며, 이를 import하기 위해서 경로가 다를 수 있기 때문에 폴더 경로를 sys.path에 추가해주어야 한다.

현재 models라는 폴더 안에 model.py 파일이 들어있는 구조이다.

import sys
sys.path.append('/home/user/file/models') # __init__.py가 있는 폴더 경로(모듈)

# 다음과 같이 경로에 해당하는 파이썬 파일 이름에서 class명 또는 함수명을 통해 import한다.
from models.model import get_pred_model
from models.model import get_seg_model

 

__init__.py파일에 필요한 함수나 클래스를 추가하여 깔끔하게 불러올수도 있다.

# __init__.py 파일에 필요한 함수 및 클래스 추가
from .model_name import get_pred_model, get_seg_model

다음과 같이 추가한 이후에는 

from models import get_pred_model
from models import get_seg_model

다음과 같이 불러올 수 있다.

 

__init__.py가 없는 경우에도 파이썬 파일을 직접 임포트해서 사용한다.

파이썬 파일명을 사용하는 것은 동일하며 현재 model.py와 camera.py가 common 폴더 안에 있는 구조이다.

import sys
sys.path.append('/home/user/file/models') # __init__.py가 있는 폴더 경로(모듈)

from common.model import TemporalModel 
from common.camera import normalize_screen_coordinates

 

다음과 같이 import하여 다양한 모듈을 불러와 사용할 수 있다.