SamsungLabs / rome

Realistic mesh-based avatars. ECCV 2022
Other
428 stars 41 forks source link

ModuleNotFoundError: No module named 'src' #4

Closed osushilover closed 2 years ago

osushilover commented 2 years ago

There may be some elementary problems with this ipynb. I run this cell on Colab Pro.

`import os, sys import importlib, warnings import argparse from glob import glob

from torchvision import transforms from tqdm import tqdm from PIL import Image

import cv2 import numpy as np import torch import matplotlib.pyplot as plt

os.chdir('..')

from src.utils import args as args_utils from src.utils.processing import process_black_shape, tensor2image from src.utils.visuals import mask_errosion

warnings.filterwarnings("ignore")

%load_ext autoreload %autoreload 2`

And then, Here is the output said🧐

ModuleNotFoundError Traceback (most recent call last) in () 15 os.chdir('..') 16 ---> 17 from src.utils import args as args_utils 18 from src.utils.processing import process_black_shape, tensor2image 19 from src.utils.visuals import mask_errosion

ModuleNotFoundError: No module named 'src'

khakhulin commented 2 years ago

if you are on colab, then probably your working directory is not correct (by default), please change it or add the correct path for import.

khakhulin commented 2 years ago
!  git clone https://github.com/SamsungLabs/rome.git
os.chdir('rome')

It will solve your problem And be sure to remove os.chdir('..')