20 lines
762 B
Python
20 lines
762 B
Python
CAST_TYPES = ["CAST", "UCAST"]
|
|
CAST_DEFAULT = "CAST"
|
|
|
|
CAST_NET_AE_PATH = "latest_net_AE.pth"
|
|
CAST_NET_DEC_B_PATH = "latest_net_Dec_B.pth"
|
|
CAST_VGG_PATH = "models/vgg_normalised.pth"
|
|
|
|
EFDM_STYLE_TYPES = ["adain", "adamean", "adastd", "efdm", "hm"]
|
|
EFDM_DEFAULT = "efdm"
|
|
EFDM_PATH = "models/hm_decoder_iter_160000.pth"
|
|
|
|
MICROAST_CONTENT_ENCODER_PATH = "models/microast/content_encoder_iter_160000.pth.tar"
|
|
MICROAST_STYLE_ENCODER_PATH = "models/microast/style_encoder_iter_160000.pth.tar"
|
|
MICROAST_DECODER_PATH = "models/microast/decoder_iter_160000.pth.tar"
|
|
MICROAST_MODULATOR_PATH = "models/microast/modulator_iter_160000.pth.tar"
|
|
|
|
UNIST_PATH = "models/unist/UniST_model.pt"
|
|
UNIST_DEC_PATH = "models/unist/dec_r41.pth"
|
|
UNIST_ENC_PATH = "models/unist/vgg_r41.pth"
|