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