diff --git a/decalib/datasets/detectors.py b/decalib/datasets/detectors.py index 631cd400..0b0ff05f 100755 --- a/decalib/datasets/detectors.py +++ b/decalib/datasets/detectors.py @@ -19,7 +19,7 @@ class FAN(object): def __init__(self): import face_alignment - self.model = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=False) + self.model = face_alignment.FaceAlignment(face_alignment.LandmarksType.TWO_D, flip_input=False) def run(self, image): ''' diff --git a/decalib/utils/renderer.py b/decalib/utils/renderer.py index 6762c2b0..94264e6b 100755 --- a/decalib/utils/renderer.py +++ b/decalib/utils/renderer.py @@ -38,7 +38,7 @@ def set_rasterizer(type = 'pytorch3d'): standard_rasterize_cuda = \ load(name='standard_rasterize_cuda', sources=[f'{curr_dir}/rasterizer/standard_rasterize_cuda.cpp', f'{curr_dir}/rasterizer/standard_rasterize_cuda_kernel.cu'], - extra_cuda_cflags = ['-std=c++14', '-ccbin=$$(which gcc-7)']) # cuda10.2 is not compatible with gcc9. Specify gcc 7 + extra_cuda_cflags = ['-std=c++14', '-ccbin=$$(which gcc-11)']) # cuda10.2 is not compatible with gcc9. Specify gcc 7 from standard_rasterize_cuda import standard_rasterize # If JIT does not work, try manually installation first # 1. see instruction here: pixielib/utils/rasterizer/INSTALL.md diff --git a/requirements.txt b/requirements.txt index 455f1732..99223f19 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,12 +6,12 @@ chumpy>=0.69 scikit-image>=0.15 opencv-python>=4.1.1 scikit-image>=0.15 #skimage -PyYAML==5.1.1 -torch==1.6.0 # for compatible with pytorch3d -torchvision==0.7.0 +PyYAML +torch==1.11.0 # for compatible with pytorch3d +torchvision==0.12.0 face-alignment yacs==0.1.8 -kornia==0.4.0 +kornia ninja fvcore # pytorch3d \ No newline at end of file