|
|
@@ -1,5 +1,5 @@
|
|
|
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.secret_func import get_secret, verify
|
|
|
|
|
|
@@ -44,3 +44,6 @@ if __name__ == '__main__':
|
|
|
print(label)
|
|
|
print(len(label))
|
|
|
print(label == secret)
|
|
|
+
|
|
|
+ # 测试验证模型准确率
|
|
|
+ compare_pred_result(result_file='./upload/006643.txt',pre_result_file='./resource/results/006643.txt')
|
