| from src.models.architectures.transformer import Encoder_TRANSFORMER, Decoder_TRANSFORMER
|
| from src.models.modeltype.motionclip import MOTIONCLIP
|
|
|
| JOINTSTYPES = ["a2m", "a2mpl", "smpl", "vibe", "vertices"]
|
|
|
| LOSSES = ["rc", "rcxyz", "vel", "velxyz"]
|
|
|
| def get_model(parameters, clip_model):
|
| encoder = Encoder_TRANSFORMER(**parameters)
|
| decoder = Decoder_TRANSFORMER(**parameters)
|
| parameters["outputxyz"] = "rcxyz" in parameters["lambdas"]
|
| return MOTIONCLIP(encoder, decoder, clip_model=clip_model, **parameters).to(parameters["device"])
|
|
|
