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-07 08:59:20 +00:00
parent 0b518e7f0e
commit e192f6f9e3
21 changed files with 1004 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sample_pro_2.py
+18 -10
View File
@@ -46,7 +46,7 @@ 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_skin_var import config as config_main
from configs.iccv.config_abl_all_skirt_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
@@ -289,7 +289,7 @@ def postprocess(mesh, body, faces, mask, do_smoothing=True):
return mesh
def sampling_example(frame, camera, nr=10):
def sampling_example(frame, camera, nr=15):
batch = get_temp_batch(frame, config_main)
mask = batch['mesh_mask'][0].numpy().astype(bool)
@@ -417,19 +417,25 @@ if __name__ == '__main__':
args = vars(parser.parse_args())
smpl = SMPL()
F = np.load("custom_utils/smpl_faces.npy")
if config_main['is_skirt']:
F = np.load("custom_utils/smpl_faces_skirt.npy")
folder = os.path.join('/home/workspace/local-data_128/train_skirt')
else:
F = np.load("custom_utils/smpl_faces.npy")
folder = os.path.join('/home/workspace/local-data/train')
# F = np.load("custom_utils/smpl_faces_skirt.npy")
F_body = np.load("custom_utils/smpl_faces.npy")
# folder = os.path.join('/home/workspace/data/dataset', ('test_skirt_ready_2' if is_skirt else 'test_ready'))
# folder = os.path.join('/home/workspace/local-data/train_skirt')
folder = os.path.join('/home/workspace/local-data/train')
model = CVAE(config_main)
curv_model = CVAECurv(config_curv)
# curv_model = CVAECurv(config_curv)
load_model(model, 'checkpoints/iccv/' + config_main['name'])
load_model(curv_model, 'checkpoints/iccv/' + config_curv['name'])
# load_model(curv_model, 'checkpoints/iccv/' + config_curv['name'])
camera_1 = dict(up=dict(x=0, y=0, z=1), center=dict(x=0, y=0, z=-0.05), eye=dict(x=0.3, y=1.0, z=0.05))
# faces_folder = '/home/workspace/data/results/support/faces'
@@ -464,10 +470,12 @@ if __name__ == '__main__':
# frame = '04925_272'
# frames = ['01070_13', '02276_232', '02978_242', '04008_295', '04173_193']
# out_folder = 'examples/sampling'
# os.makedirs(out_folder, exist_ok=True)
# for frame in tqdm.tqdm(frames):
# sampling_example(frame, camera_1)
frames = ['00214_111', '01661_210', '02144_249', '02151_37', '02930_275', '03812_3']
out_folder = 'examples/sampling_skirt'
os.makedirs(out_folder, exist_ok=True)
for frame in tqdm.tqdm(frames):
sampling_example(frame, camera_1)
# sampling_poses(frame, camera_01082_109)
# sampling_garment(frame, camera_1)