update unpose limits

unpose_skirt.py

@@ -12,7 +12,7 @@ from scipy.spatial.transform import Rotation
 from tqdm import tqdm
 
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "1"
+os.environ["CUDA_VISIBLE_DEVICES"] = "6"
 import tensorflow as tf
 gpus = tf.config.experimental.list_physical_devices('GPU')
 if gpus:
@@ -43,7 +43,7 @@ def read_garment(folder, id):
     garment_dir = os.path.join(folder, id)
 
     garments = [f.replace('.pc16','') for f in os.listdir(garment_dir) if f.find('_') == -1 and f != 'info.mat' and not f.endswith('.obj')]
-    assert len (garments) == 1
+    assert len (garments) == 1, id
         
     garment = garments[0]
 
@@ -118,9 +118,9 @@ def plot_3d(mesh_orig, mesh_pred, unposed, faces):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/train_skirt')
-    parser.add_argument("-s", "--start", help = "start of range", default=0)
-    parser.add_argument("-e", "--end", help = "end of range", default=1)
+    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/test_skirt')
+    parser.add_argument("-s", "--start", help = "start of range", default=100)
+    parser.add_argument("-e", "--end", help = "end of range", default=200)
     parser.add_argument("-n", "--name", help = "end of range", default='lhunor')
     args = parser.parse_args()
 
@@ -251,13 +251,13 @@ if __name__ == '__main__':
             if euclid_error < 0.0001:
                 break
 
-            if euclid_error < 0.0005 and i > 15000:
+            if euclid_error < 0.0005 and i > 20000:
                 break
 
-            if euclid_error < 0.001 and i > 30000:
+            if euclid_error < 0.001 and i > 40000:
                 break
 
-            if i > 40000:
+            if i > 60000:
                 break
 
         # b = np.random.choice(np.array(range(frames)))