generating more curvatures

This commit is contained in:
2023-01-28 19:12:34 +00:00
parent 632d8db815
commit a63a47d84e
4 changed files with 25 additions and 23 deletions
+9 -8
View File
@@ -444,15 +444,16 @@ if __name__ == '__main__':
frames_file = 'frames' + skirt_modifier + '_10p.txt'
neighbourhood = 10
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)]
# 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 = [frame.split('_')[0] for frame in frames]
ids = sorted(list(set(ids)))#[int(args.start):int(args.end)]
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):