colab에서 커스텀 폴더/패키지 import 방법

 

 

colab에서 프로젝트를 진행하면 한번쯤은 겪어봤을 수 있는 이슈이다.

 

 

---

 

상황 :

 

아래와 같은 코드에서

 

- torch 는 내가 만든 패키지가 아니다 pip로 설치가능

- LFW는 내가 만든 data폴더의 lfw.py 에 있는 class이다.

 

import torch
import torch.nn as nn

from data.lfw import LFW
from data.ms1m import get_train_loader

...

하지만 같은 위치에 있다고 해도 LFW를 바로 불러오지는 못한다.

 

PC와 다르게 이와 같은 저장소의 문제가 있다.

 

 

---

 

 

해결 : 

 

매우 간단하다. 

 

- google drive 와 연동시켜준후

- 위치를 현재위치로 이동시켜주면 된다.

 

1. drive 마운트

import os
from google.colab import drive
drive.mount('/content/drive/')

 

2. 위치를 현재 위치로 이동

%cd drive/MyDrive/ColabNotebooks/Arcface

 

 

PC와 동일하게 로컬 패키지를 가져올 수 있다.

 

 

끝