luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

generate remasked data sample

Browse Source
This commit is contained in:
luni
2023-06-01 16:40:05 +00:00
parent daf51b4dbe
commit 49ef5798a5
2 changed files with 18 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/calculate_normals.py
+2 -2
View File
@@ -62,8 +62,8 @@ def get_frames(folder):
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-f", "--folder", help = "end of range", default='') parser.add_argument("-f", "--folder", help = "end of range", default='/data-net/dataset/00761')
parser.add_argument("-f_obj", "--folder_object", help = "end of range", default='/') parser.add_argument("-f_obj", "--folder_object", help = "end of range", default='/data-net/dataset/cloth3d/train')
parser.add_argument("-s", "--start", help = "start of range", default=0) parser.add_argument("-s", "--start", help = "start of range", default=0)
parser.add_argument("-e", "--end", help = "end of range", default=10) parser.add_argument("-e", "--end", help = "end of range", default=10)
parser.add_argument("-n", "--name", help = "end of range", default='lhunor') parser.add_argument("-n", "--name", help = "end of range", default='lhunor')
data/investigate.py
+16 -12
View File
@@ -242,22 +242,25 @@ def check(folder, frame):
def remask(folder, frame, mask, out_folder): def remask(folder, frame, mask, out_folder):
id, frame_nr, garment = frame.split() id, frame_nr = frame.split('_')
frame_nr = int(frame_nr) frame_nr = int(frame_nr)
info = loadInfo(os.path.join(folder, id, 'info.mat'))
if frame_nr == 0: if frame_nr == 0:
mesh_static = read_static_garment(folder, id, garment) mesh_static = np.load(os.path.join(folder, 'mesh_static', f'{id}.npz'))['mesh_static']
mesh_static_new = np.zeros((14475,3), dtype=np.float32)
mesh_static_new[mask,:] = mesh_static[mask,:]
np.savez_compressed(os.path.join(out_folder, 'mesh_static', f'{id}.npz'), mesh_static=mesh_static_new)
mesh_posed = read_posed_garment_frame(folder, id, garment, frame_nr, mask=None, shape=None, pose=info['poses'][:,frame_nr], deorient=True, unshape=None) # mesh_posed = read_posed_garment_frame(folder, id, garment, frame_nr, mask=None, shape=None, pose=info['poses'][:,frame_nr], deorient=True, unshape=None)
mesh_posed = mesh_posed[mask,:] mesh_posed = np.load(os.path.join(folder, 'mesh_posed_no_orient', f'{id}_{frame_nr}.npz'))['mesh_posed_no_orient']
os.makedirs(os.path.join(out_folder, 'mesh_posed_no_orient'), exist_ok=True) mesh_posed_new = np.zeros((14475,3), dtype=np.float32)
mesh_posed_new[mask,:] = mesh_posed[mask,:]
np.savez_compressed(os.path.join(out_folder, 'mesh_posed_no_orient', f'{id}_{frame_nr}.npz'), mesh_posed_no_orient=mesh_posed) np.savez_compressed(os.path.join(out_folder, 'mesh_posed_no_orient', f'{id}_{frame_nr}.npz'), mesh_posed_no_orient=mesh_posed)
if __name__ == "__main__": if __name__ == "__main__":
folder = r'D:\Work\Dataset\cloth3d\train' folder = '/data-local/127/train'
frames = [f'00761_{i}' for i in range(300)] frames = [f'00761_{i}' for i in range(300)]
root = '/'.join(folder.split('/')[:-1]) root = '/'.join(folder.split('/')[:-1])
type = folder.split('/')[-1] type = folder.split('/')[-1]
@@ -270,12 +273,13 @@ if __name__ == "__main__":
orient_mod = '_no_orient' if use_no_orient else '' orient_mod = '_no_orient' if use_no_orient else ''
shape_mod = '_no_shape' if no_shape else '' shape_mod = '_no_shape' if no_shape else ''
mask = np.load('/data-net/dataset/00761/mesh_mask/00761.npy')
faces = np.load('/data-net/dataset/00761/faces/00761.npy')
while True: for frame in frames:
frame = random.choice(frames) # frame = random.choice(frames)
frame = '00761 0 Tshirt' # frame = '00761 0 Tshirt'
mask, faces = check(folder, frame) remask(folder, frame, mask, '/data-net/dataset/00761')
remask(folder, frame, mask, 'temp/00761')
pass pass