From 3727f341a7d2a44f83e5538cb7465dcf75e6caff Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Fri, 7 Oct 2022 21:56:23 +0000 Subject: [PATCH] update unpose limits --- unpose_skirt.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/unpose_skirt.py b/unpose_skirt.py index 21309f6..aee8cff 100644 --- a/unpose_skirt.py +++ b/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)))