'Path'에 해당되는 글 1건

Pycharm 을 이용해 개발하다 보면 파이썬 코드가 여기저기 생기고 각 파이썬 파일마다 main 함수가 있을 것이다.

그런데 각 파이썬 파일들이 하나의 동일한  파일을 참조하게 되는 경우 (tensorflow model, sample image 등) 각 파이썬 파일마다 경로를 일일이 수정해줘야 하는 불편함이 있다.

다른 방법은 생각해보지 않았지만 그냥 단순하게 프로젝트 폴더를 root 로 삼으면 모든 파일에 대해 동일한 경로 값으로 접근이 가능하다.

get_root_dir.py



from pathlib import Path

def get_project_root() -> Path:
    return str(Path(__file__).parent.parent) + '/'

이렇게 프로젝트의 root 경로를 return 하는 간단한 파일을 작성해두고 import 해서 사용하면 각 파이썬 코드마다 참조하는 파일들의 경로를 동일하게 줄 수 있다.

아래는 예시 코드

from get_root_dir import get_project_root

ROOT_DIR = get_project_root()


model_path = ROOT_DIR + "/models/frozen_inference_graph.pb"
labelmap_path = ROOT_DIR + "/models/labelmap.pbtxt"