add CAST

2024-08-13 00:21:53 +07:00
parent d13c7270ae
commit 0e39efc7c8
13 changed files with 1360 additions and 25 deletions
@@ -0,0 +1,6 @@
+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"