| import os
|
| from .base import ArgumentParser, adding_cuda
|
| from .tools import load_args
|
|
|
|
|
| def parser():
|
| parser = ArgumentParser()
|
| parser.add_argument("checkpointname")
|
|
|
| opt = parser.parse_args()
|
|
|
| folder, checkpoint = os.path.split(opt.checkpointname)
|
| parameters = load_args(os.path.join(folder, "opt.yaml"))
|
|
|
| adding_cuda(parameters)
|
| epoch = int(checkpoint.split("_")[-1].split('.')[0])
|
| return parameters, folder, checkpoint, epoch
|
|
|
|
|
| def construct_checkpointname(parameters, folder):
|
| implist = [parameters["modelname"],
|
| parameters["dataset"],
|
| parameters["extraction_method"],
|
| parameters["pose_rep"]]
|
| if parameters["pose_rep"] != "xyz":
|
|
|
| if "glob" in parameters:
|
| implist.append("glob" if parameters["glob"] in [True, ""] else "noglob")
|
| else:
|
| implist.append("noglob")
|
| if "translation" in parameters:
|
| implist.append("translation" if parameters["translation"] in [True, ""] else "notranslation")
|
| else:
|
| implist.append("notranslation")
|
|
|
| if "rcxyz" in parameters["modelname"]:
|
| implist.append("joinstype_{}".format(parameters["jointstype"]))
|
|
|
| if "num_layers" in parameters:
|
| implist.append("numlayers_{}".format(parameters["num_layers"]))
|
|
|
| for name in ["num_frames", "min_len", "max_len", "num_seq_max"]:
|
| pvalue = parameters[name]
|
| pname = name.replace("_", "")
|
| if pvalue != -1:
|
| implist.append(f"{pname}_{pvalue}")
|
|
|
| if "view" in parameters:
|
| if parameters["view"] == "frontview":
|
| implist.append("frontview")
|
|
|
| if "use_z" in parameters:
|
| if parameters["use_z"] != 0:
|
| implist.append("usez")
|
| else:
|
| implist.append("noz")
|
|
|
| if "vertstrans" in parameters:
|
| implist.append("vetr" if parameters["vertstrans"] else "novetr")
|
|
|
| if "ablation" in parameters:
|
| abl = parameters["ablation"]
|
| if abl not in ["", None]:
|
| implist.append(f"abl_{abl}")
|
|
|
| if parameters["num_frames"] != -1:
|
| implist.append("sampling_{}".format(parameters["sampling"]))
|
| if parameters["sampling"] == "conseq":
|
| implist.append("samplingstep_{}".format(parameters["sampling_step"]))
|
| if "lambda_kl" in parameters:
|
| implist.append("kl_{:.0e}".format(float(parameters["lambda_kl"])))
|
|
|
| if "activation" in parameters:
|
| act = parameters["activation"]
|
| implist.append(act)
|
|
|
| implist.append("bs_{}".format(parameters["batch_size"]))
|
| implist.append("ldim_{}".format(parameters["latent_dim"]))
|
|
|
| checkpoint = "_".join(implist)
|
| return os.path.join(folder, checkpoint)
|
|
|
|
|
|
|
