Prechádzať zdrojové kódy

修改YOLOX模型白盒水印嵌入投影矩阵保存位置

liyan 10 mesiacov pred
rodič
commit
dcdf5dbc2a

+ 1 - 1
watermark_generate/deals/yolox_pytorch_white_embed.py

@@ -139,7 +139,7 @@ f"""
                 if isinstance(base_conv, BaseConv):
                     conv_layers.append(base_conv.conv)
         conv_layers = conv_layers[0:2]
-        self.encoder = ModelEncoder(layers=conv_layers, secret=secret_label, key_path='./YOLOX_outputs/key.pt', device='cuda')
+        self.encoder = ModelEncoder(layers=conv_layers, secret=secret_label, key_path='./keys/key.npy', device='cuda')
     """
     # 向工程文件追加函数
     modify_file.append_block_in_file(project_file, append_source_block)