fixed compatibility issues with python 3.10

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):
         '''

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

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