27 lines
1.0 KiB
Python
27 lines
1.0 KiB
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"
|
|
|
|
VGG_NORM_CONV51_PATH = "models/vgg_normalised_conv5_1.pth"
|
|
AESPA_DEC_PATH = "models/aespa/dec_model.pth"
|
|
AESPA_TRANSFORMER_PATH = "models/aespa/transformer_model.pth"
|
|
|
|
TSSAT_DEC_PATH = "models/tssat/decoder_iter_160000.pth"
|
|
|
|
AESFA_PATH = "models/aesfa/main.pth" |