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

use unregistered input

Browse Source
This commit is contained in:
luni
2023-01-23 12:29:10 +00:00
parent ab78dab165
commit a0f3e95255
4 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/preprocess.py
+5 -6
View File
@@ -30,12 +30,11 @@ def generate_frames_file(folder, file_name):
garment_dir = os.path.join(folder, id)
if not os.path.isdir(garment_dir):
continue
garments = [f.replace('_registered.pc16','') for f in os.listdir(garment_dir) if f.endswith('_registered.pc16') and not f.startswith('Combined')]
assert len(garments) == 1
garment = garments[0]
verts = readPC2(os.path.join(garment_dir, garment + '_registered.pc16'), True)['V']
for i in range(verts.shape[0]):
frames.append(Frame(id, i, garment))
garments = [f.replace('.obj','') for f in os.listdir(garment_dir) if f.endswith('.obj')]
info = loadInfo(os.path.join(garment_dir, 'info'))
garments_str = ' '.join(garments)
for i in range(info['poses'].shape[1]):
frames.append(Frame(id, i, garments_str))
with open(os.path.join(folder, file_name), 'w') as f:
for i, frame in enumerate(frames):