fixed bug in import

amosgropp · Jan 23, 2021 · cded34e · cded34e
1 parent 4b72dc0
commit cded34e
Show file tree

Hide file tree

Showing 6 changed files with 24 additions and 5 deletions.
diff --git a/code/preprocess/dfaust.py b/code/preprocess/dfaust.py
@@ -1,3 +1,8 @@
+import os
+import sys
+project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+sys.path.append(project_dir)
+os.chdir(project_dir)
 from __future__ import print_function
 import argparse
 
@@ -93,4 +98,4 @@
 
                         np.save(output_file + '_normalization.npy', {"center":center,"scale":scale})
 
-    print ("end!")
+    print ("end!")
diff --git a/code/reconstruction/run.py b/code/reconstruction/run.py
@@ -3,7 +3,6 @@
 project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
 sys.path.append(project_dir)
 os.chdir(project_dir)
-import os
 from datetime import datetime
 from pyhocon import ConfigFactory
 import numpy as np

diff --git a/code/shapespace/eval.py b/code/shapespace/eval.py
@@ -1,5 +1,9 @@
 import argparse
 import os
+import sys
+project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+sys.path.append(project_dir)
+os.chdir(project_dir)
 import json
 import utils.general as utils
 import torch

diff --git a/code/shapespace/interpolate.py b/code/shapespace/interpolate.py
@@ -1,5 +1,9 @@
-import argparse
 import os
+import sys
+project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+sys.path.append(project_dir)
+os.chdir(project_dir)
+import argparse
 import json
 import utils.general as utils
 import torch

diff --git a/code/shapespace/latent_optimizer.py b/code/shapespace/latent_optimizer.py
@@ -1,3 +1,8 @@
+import os
+import sys
+project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+sys.path.append(project_dir)
+os.chdir(project_dir)
 import torch
 from model.network import gradient
 from model.sample import Sampler
@@ -66,4 +71,4 @@ def optimize_latent(points, normals, conf, num_of_iterations, network, lr=1.0e-2
 
         print('latent loss iter {0}:{1}'.format(i, loss.item()))
 
-    return latent.unsqueeze(0)
+    return latent.unsqueeze(0)
diff --git a/code/shapespace/train.py b/code/shapespace/train.py
@@ -1,6 +1,8 @@
 import os
 import sys
-import os
+project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
+sys.path.append(project_dir)
+os.chdir(project_dir)
 from datetime import datetime
 from pyhocon import ConfigFactory
 from time import time