Rename pa_desc_* weights to ninja_desc_*

Renames the logical weight identifiers (PA_DESC_MODEL dicts,
download_weights() calls, README pretrained-models table) and
the underlying torchscript filenames.

Author: hyojinie

README.md

@@ -45,10 +45,10 @@ Weights are released as GitHub Release assets. They are downloaded automatically
 
 | Logical name                                       | File                                                       |
 | -------------------------------------------------- | ---------------------------------------------------------- |
-| `pa_desc_sosnet_init`                              | `pa_desc_sos_init_torchscript.pt`                          |
-| `pa_desc_sift_init`                                | `pa_desc_sift_init_torchscript.pt`                         |
-| `pa_desc_hardnet_init`                             | `pa_desc_hardnet_init_torchscript.pt`                      |
-| `pa_desc_hardnet_init_normalize`                   | `pa_desc_hardnet_init_normalize_torchscript.pt`            |
+| `ninja_desc_sosnet_init`                           | `ninja_desc_sos_init_torchscript.pt`                       |
+| `ninja_desc_sift_init`                             | `ninja_desc_sift_init_torchscript.pt`                      |
+| `ninja_desc_hardnet_init`                          | `ninja_desc_hardnet_init_torchscript.pt`                   |
+| `ninja_desc_hardnet_init_normalize`                | `ninja_desc_hardnet_init_normalize_torchscript.pt`         |
 | `lemuria_unet_padesc_sos_init`                     | `LemuriaNet_Unet_PADesc_INIT.pth`                          |
 | `lemuria_unet_padesc_sift_init`                    | `LemuriaNet_UNet_PADesc_SIFT_INIT.pth`                     |
 | `lemuria_unet_padesc_hardnet_init`                 | `LemuriaNet_UNet_PADesc_HARDNET_INIT.pth`                  |

ninjadesc/_compat/weights.py

@@ -12,10 +12,10 @@
 _RELEASE_BASE = "https://github.com/facebookresearch/ninjadesc/releases/download/v1.0"
 
 WEIGHT_FILES = {
-    "pa_desc_sosnet_init": "pa_desc_sos_init_torchscript.pt",
-    "pa_desc_sift_init": "pa_desc_sift_init_torchscript.pt",
-    "pa_desc_hardnet_init": "pa_desc_hardnet_init_torchscript.pt",
-    "pa_desc_hardnet_init_normalize": "pa_desc_hardnet_init_normalize_torchscript.pt",
+    "ninja_desc_sosnet_init": "ninja_desc_sos_init_torchscript.pt",
+    "ninja_desc_sift_init": "ninja_desc_sift_init_torchscript.pt",
+    "ninja_desc_hardnet_init": "ninja_desc_hardnet_init_torchscript.pt",
+    "ninja_desc_hardnet_init_normalize": "ninja_desc_hardnet_init_normalize_torchscript.pt",
     "lemuria_unet_padesc_sos_init": "LemuriaNet_Unet_PADesc_INIT.pth",
     "lemuria_unet_padesc_sift_init": "LemuriaNet_UNet_PADesc_SIFT_INIT.pth",
     "lemuria_unet_padesc_hardnet_init": "LemuriaNet_UNet_PADesc_HARDNET_INIT.pth",

ninjadesc/pa_desc/engine/joint_knob_module.py

@@ -23,7 +23,7 @@
 
 
 DISC_ON = 0
-PA_DESC_MODEL = "pa_desc_sosnet_init"
+PA_DESC_MODEL = "ninja_desc_sosnet_init"
 RECON_MODEL = "lemuria_unet_padesc_sos_init"
 
 

ninjadesc/pa_desc/engine/joint_module.py

@@ -26,9 +26,9 @@
 
 
 PA_DESC_MODEL = {
-    "sosnet": "pa_desc_sosnet_init",
-    "sift": "pa_desc_sift_init",
-    "hardnet": "pa_desc_hardnet_init",
+    "sosnet": "ninja_desc_sosnet_init",
+    "sift": "ninja_desc_sift_init",
+    "hardnet": "ninja_desc_hardnet_init",
 }
 
 RECON_MODEL = {

ninjadesc/pa_desc/engine/lemurianet_module.py

@@ -27,9 +27,9 @@
     "sift": "lemuria_unet_sift",
 }
 PA_DESC_MODEL = {
-    "sosnet": "pa_desc_sosnet_init",
-    "sift": "pa_desc_sift_init",
-    "hardnet": "pa_desc_hardnet_init",
+    "sosnet": "ninja_desc_sosnet_init",
+    "sift": "ninja_desc_sift_init",
+    "hardnet": "ninja_desc_hardnet_init",
 }
 
 

ninjadesc/pa_desc/engine/padesc_module.py

@@ -38,7 +38,7 @@ def __init__(self, cfg: DictConfig):
             self.base_desc = torch.jit.load(download_weights("hardnet_lib"))
             self.base_desc.eval()
 
-        padesc_model = torch.jit.load(download_weights("pa_desc_sosnet_init"))
+        padesc_model = torch.jit.load(download_weights("ninja_desc_sosnet_init"))
         self.privacy = padesc_model.privacy
         self.base_desc_type = cfg.base_desc