ソースを参照

修改faster-rcnn模型黑盒水印检测流程,并修改脚本文件名称

liyan 1 ヶ月 前
コミット
c0cfb606ad

+ 2 - 2
README.md

@@ -52,8 +52,8 @@ model_watermark_detect
         │   ├── classification_all_whitebox_process.py
         │   ├── classification_pytorch_blackbox_process.py
         │   ├── classification_tensorflow_blackbox_process.py
-        │   ├── faster-rcnn_pytorch_blackbox_process.py
-        │   ├── faster-rcnn_pytorch_whitebox_process.py
+        │   ├── faster_rcnn_pytorch_blackbox_process.py
+        │   ├── faster_rcnn_pytorch_whitebox_process.py
         │   ├── general_process_define.py
         │   ├── googlenet_all_whitebox_process.py
         │   ├── ssd_pytorch_blackbox_process.py

+ 2 - 2
watermark_verify/process/faster-rcnn_pytorch_blackbox_process.py

@@ -100,8 +100,8 @@ def detect_watermark(results, watermark_box, threshold=0.5):
     # 解析输出结果
     if len(results[0]) == 0:
         return False
-    top_label = np.array(results[0][:, 4], dtype='int32')
-    top_conf = results[0][:, 5]
+    top_label = np.array(results[0][:, 5], dtype='int32')
+    top_conf = results[0][:, 4]
     top_boxes = results[0][:, :4]
     for box, score, cls in zip(top_boxes, top_conf, top_label):
         wm_box_coords = watermark_box[:4]

watermark_verify/process/faster-rcnn_pytorch_whitebox_process.py → watermark_verify/process/faster_rcnn_pytorch_whitebox_process.py