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

save

Browse Source
This commit is contained in:
luni
2023-03-10 17:25:44 +00:00
parent cb76d36c4f
commit 4b1cb0b06d
27 changed files with 1890 additions and 207 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sample_pro_2.py
+13 -7
View File
@@ -46,7 +46,8 @@ from models.pyramid_model_optimized import Pyramid
# from configs.config_abl_overfit_dress_var_3 import config as config_main
# from configs.config_abl_overfit_jumpsuit_2 import config as config_main
# from configs.config_cvae_curv import config as config_curv
from configs.iccv.config_abl_all_skirt_var import config as config_main
# from configs.iccv.config_abl_all_skirt_var import config as config_main
from configs.iccv.config_abl_skin_var import config as config_main
from configs.iccv.config_cvae_curv import config as config_curv
# from models.cvae_model_overfit_jumpsuit import CVAE
from models.cvae_model import CVAENew as CVAE
@@ -331,11 +332,16 @@ def sampling_example(frame, camera, nr=15):
for i in range(nr):
plot_fixed(posed_mesh_fixed[i], body, faces, camera).write_image(os.path.join(out_folder, frame, frame + '_' + str(i) + '.png'))
save_stl(os.path.join(out_folder, frame, frame + '_body.stl'), body, F_body)
save_stl(os.path.join(out_folder, frame, frame + '_gt.stl'), gt_mesh, faces.astype(np.int32))
# save_stl(os.path.join(out_folder, frame, frame + '_body.stl'), body, F_body)
# save_stl(os.path.join(out_folder, frame, frame + '_gt.stl'), gt_mesh, faces.astype(np.int32))
# for i in range(nr):
# save_stl(os.path.join(out_folder, frame, frame + '_' + str(i) + '.stl'), posed_mesh_fixed[i], faces.astype(np.int32))
np.save(os.path.join(out_folder, frame, frame + '_gt.npy'), gt_mesh)
np.save(os.path.join(out_folder, frame, frame + '_body.npy'), body)
np.save(os.path.join(out_folder, frame, frame + '_faces.npy'), faces.astype(np.int32))
for i in range(nr):
save_stl(os.path.join(out_folder, frame, frame + '_' + str(i) + '.stl'), posed_mesh_fixed[i], faces.astype(np.int32))
np.save(os.path.join(out_folder, frame, frame + '_' + str(i) + '.npy'), posed_mesh_fixed[i])
# custom_imshow(np.concatenate(row_sampling, axis=1)).write_image(os.path.join(out_folder, frame + '_sampling.png'))
print()
@@ -470,9 +476,9 @@ if __name__ == '__main__':
# frame = '04925_272'
# frames = ['01070_13', '02276_232', '02978_242', '04008_295', '04173_193']
frames = ['00214_111', '01661_210', '02144_249', '02151_37', '02930_275', '03812_3']
frames = ['02276_232', '02978_242', '04173_193']
out_folder = 'examples/sampling_skirt'
out_folder = 'examples/sampling_analysis'
os.makedirs(out_folder, exist_ok=True)
for frame in tqdm.tqdm(frames):
sampling_example(frame, camera_1)