Update modelsRig.py

cassiePython · Jul 18, 2022 · 4660a56 · 4660a56
1 parent 3b63d28
commit 4660a56
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/models/modelsRig.py b/models/modelsRig.py
@@ -231,13 +231,13 @@ def forward_test(self, latent_w, pca_id, pca_val):
             if self._model_name == 'RIGModelS':     
                 What = self.RigNetDecoder.forwardShape(I, Pv, latent_w, Pw)
             elif self._model_name == 'RIGModelE':     
-                What = self.RigNetDecoder.forwardExpression(I, Pv, latent, Pw)
+                What = self.RigNetDecoder.forwardExpression(I, Pv, latent_w, Pw)
             elif self._model_name == 'RIGModelP':     
-                What = self.RigNetDecoder.forwardPose(I, Pv, latent, Pw)
+                What = self.RigNetDecoder.forwardPose(I, Pv, latent_w, Pw)
             elif self._model_name == 'RIGModelL':     
-                What = self.RigNetDecoder.forwardLight(I, Pv, latent, Pw)
+                What = self.RigNetDecoder.forwardLight(I, Pv, latent_w, Pw)
             elif self._model_name == 'RIGModelA':     
-                What = self.RigNetDecoder.forwardAlbedo(I, Pv, latent, Pw)
+                What = self.RigNetDecoder.forwardAlbedo(I, Pv, latent_w, Pw)
 
             WhatF = What.view(1, -1)
             Pwhat = self.APNet(WhatF)