Browse Source

修改测试代码

liyan 1 năm trước cách đây
mục cha
commit
e9697ee6af
1 tập tin đã thay đổi với 4 bổ sung1 xóa
  1. 4 1
      tests/test_gen_qrcodes.py

+ 4 - 1
tests/test_gen_qrcodes.py

@@ -1,5 +1,5 @@
 from watermark_generate.tools.dataset_process import process_train_dataset, generate_trigger_dataset, \
 from watermark_generate.tools.dataset_process import process_train_dataset, generate_trigger_dataset, \
-    extract_crypto_label_from_trigger
+    extract_crypto_label_from_trigger, compare_pred_result
 from watermark_generate.tools.gen_qrcodes import generate_qrcodes, detect_qrcode_in_bbox, extract_qrcode_from_image
 from watermark_generate.tools.gen_qrcodes import generate_qrcodes, detect_qrcode_in_bbox, extract_qrcode_from_image
 from watermark_generate.tools.secret_func import get_secret, verify
 from watermark_generate.tools.secret_func import get_secret, verify
 
 
@@ -44,3 +44,6 @@ if __name__ == '__main__':
     print(label)
     print(label)
     print(len(label))
     print(len(label))
     print(label == secret)
     print(label == secret)
+
+    # 测试验证模型准确率
+    compare_pred_result(result_file='./upload/006643.txt',pre_result_file='./resource/results/006643.txt')