| | 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)
|
| |
|
| |
|
| |
|
