pipeline.py 等更新

2025-01-21 18:22:56 +08:00
parent bfe7bc0fd5
commit 64248b1557
15 changed files with 106 additions and 77 deletions
--- a/contrast/feat_extract/config.py
+++ b/contrast/feat_extract/config.py
@ -61,8 +61,9 @@ class Config:
    test_val = "D:/比对/cl"
    # test_val = "./data/test_data_100"
    
-    test_model = "checkpoints/best_resnet18_v12.pth"   
+    # test_model = "checkpoints/best_resnet18_v12.pth"   
    # test_model = "checkpoints/zhanting_res_801.pth"
+    test_model = "checkpoints/zhanting_res_abroad_8021.pth"