eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework
https://eladrich.github.io/pixel2style2pixel/
MIT License
3.2k stars 568 forks source link

[bug] Colab notebook missing dep. upfirdn2d.so #185

Closed philipwilk closed 3 years ago

philipwilk commented 3 years ago

The last cell before selecting the experiment type is missing dep upfirdn2d.so where it imports pSp;

from argparse import Namespace
import time
import sys
import pprint
import numpy as np
from PIL import Image
import torch
import torchvision.transforms as transforms

sys.path.append(".")
sys.path.append("..")

from datasets import augmentations
from utils.common import tensor2im, log_input_image
from models.psp import pSp

%load_ext autoreload
%autoreload 2

image

yuval-alaluf commented 3 years ago

I just ran the notebook myself and could not reproduce the error. Are you running the notebook locally or using Google Colab?

philipwilk commented 3 years ago

I just ran the notebook myself and could not reproduce the error. Are you running the notebook locally or using Google Colab?

I ran it in Google Colab, hm. It might be a supply chain issue with one of the dependencies of a dependency. 🤷🏼‍♀️

yuval-alaluf commented 3 years ago

Definitely weird. I would make sure that all the previous cells ran successfully and that you didn't install anything else by mistake. But other than that I'm not really sure where the difference could come from 🤔

philipwilk commented 3 years ago

I'm gonna close this issue because if it can't be replicated then it should be fine 😆