update

convert_to_uv_local_single.py

@@ -384,7 +384,7 @@ def test(verts, faces, values):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/local-data/valid_skirt')
+    parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/local-cache/train_skirt')
     parser.add_argument("-s", "--start", help = "start of range", default=0)
     parser.add_argument("-e", "--end", help = "end of range", default=30000)
     parser.add_argument("-n", "--name", help = "end of range", default='lhunor')
@@ -442,17 +442,19 @@ if __name__ == '__main__':
 
     folder = args.folder
     frames_file = 'frames' + skirt_modifier + '_10p.txt'
-    neighbourhood = 10
+    neighbourhood = 100
 
-    # frames = []
-    # with open(os.path.join(folder, frames_file), 'r') as f:
-    #     lines = f.readlines()
-    #     lines = [line.strip() for line in lines]
-    # frames = lines[int(args.start):int(args.end)]
-    # ids = [frame.split('_')[0] for frame in frames]
+    frames = []
+    with open(os.path.join(folder, frames_file), 'r') as f:
+        lines = f.readlines()
+        lines = [line.strip() for line in lines]
+    frames = lines#[int(args.start):int(args.end)]
+    ids = [frame.split('_')[0] for frame in frames]
 
-    ids = os.listdir(os.path.join(local_folder, 'pose_shape'))
-    ids = [id.split('_')[0] for id in ids]
+    print(f'params: -f {args.folder} -s {args.start} -e {args.end} -n {args.name} -skirt {args.skirt} -neigh {neighbourhood}')
+
+    # ids = os.listdir(os.path.join(local_folder, 'pose_shape'))
+    # ids = [id.split('_')[0] for id in ids]
     ids = sorted(list(set(ids)))[int(args.start):int(args.end)]
 
     for id in tqdm.tqdm(ids):