enable texture correction

Author: aashishrai3799

decalib/deca.py

@@ -440,12 +440,12 @@ def decode(self, codedict, name, rendering=True, iddict=None, vis_lmk=True, retu
 
             ### Texture Correction
 
-            '''angle1 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,3555,2205])
+            angle1 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,3555,2205])
             angle2 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,723,3555])
             avg_ang = int((angle1+angle2)/2)
             avg_ang = 90-(360-avg_ang)
             correct_tex = tex_correction(uv_texture_gt[0].permute(1,2,0).detach().cpu(), avg_ang)
-            uv_texture_gt = correct_tex.permute(2,0,1)[None,...].to('cuda')'''
+            uv_texture_gt = correct_tex.permute(2,0,1)[None,...].to('cuda')
 
             # uv_shading = self.render.add_SHlight(uv_detail_normals, codedict['light'])
 
@@ -537,13 +537,13 @@ def decode(self, codedict, name, rendering=True, iddict=None, vis_lmk=True, retu
 
                     ### Texture Correction
 
-                    '''angle1 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,3555,2205])
+                    angle1 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,3555,2205])
                     angle2 = mesh_angle(verts[0].detach().cpu().numpy(), [3572,723,3555])
                     avg_ang = int((angle1+angle2)/2)
                     avg_ang = 90-(360-avg_ang)
                     correct_tex = tex_correction(uv_texture_gt[0].permute(1,2,0).detach().cpu(), avg_ang)
                     uv_texture_gt = correct_tex.permute(2,0,1)[None,...].to('cuda')
-                    uv_texture_gt = uv_texture_gt[:,:3,:,:]*self.uv_face_eye_mask + (uv_texture[:,:3,:,:]*(1-self.uv_face_eye_mask))'''
+                    uv_texture_gt = uv_texture_gt[:,:3,:,:]*self.uv_face_eye_mask + (uv_texture[:,:3,:,:]*(1-self.uv_face_eye_mask))
 
 
                 else: