diff --git a/configs/post/config_abl_pyr_32.py b/configs/post/config_abl_pyr_32.py index 4836c52..9a55ddd 100644 --- a/configs/post/config_abl_pyr_32.py +++ b/configs/post/config_abl_pyr_32.py @@ -1,9 +1,9 @@ config = dict() -config['gpu'] = '2' +config['gpu'] = '6' config['name'] = 'abl_pyr_32' -config['notes'] = 'fixed noise handling' -config['tags'] = ['pyr_post'] +config['notes'] = '' +config['tags'] = ['review_1'] # DON'T FORGET TO RESET SSIM LOSS USAGE # DON'T FORGET TO CHECK MODEL LOADING diff --git a/preprocessing.py b/preprocessing.py index 4b81f54..8efb8df 100644 --- a/preprocessing.py +++ b/preprocessing.py @@ -232,17 +232,17 @@ def generate_frames_file(folder, ratio=0.1): selected = [] for id in tqdm.tqdm(sub_ids): - if garment_type_table[id] != 'Dress': - continue - if len(selected) > 30000: - break + # if garment_type_table[id] != 'Dress': + # continue + # if len(selected) > 30000: + # break # frames = glob.glob(os.path.join(folder, 'pose_shape', id + '_*')) frames = [frame for frame in all_frames if frame.startswith(id + '_')] frames = [frame.split('/')[-1].split('.')[0] for frame in frames] selected.extend(frames) - with open(os.path.join(folder, 'frames_dress_30k.txt'), 'w') as f: + with open(os.path.join(folder, 'frames_skirt_10p.txt'), 'w') as f: for i, frame in enumerate(selected): f.write(frame + '\n') @@ -293,9 +293,9 @@ def generate_frames_file_filtered_cape(): if __name__ == '__main__': - src_folder = '/home/workspace/local-data/train_skirt' #cloth3d_skirt + src_folder = '/home/workspace/local-data/valid_skirt' #cloth3d_skirt # src_folder = r'D:\Work\Dataset\validation_2' - dest_folder = '/home/workspace/data/dataset/validation' + dest_folder = '/home/workspace/data/dataset/local-cache/train_skirt' # create_no_orient_body(src_folder) @@ -306,5 +306,5 @@ if __name__ == '__main__': # verify_poses(src_folder) # create_garment_type_table() # generate_faces() - generate_frames_file(src_folder, ratio=1) + generate_frames_file(src_folder, ratio=0.1) # generate_frames_file_filtered_cape()