update swinface source code

source/swin_b_test_lfw.py

@@ -172,10 +172,10 @@ def evaluate_test_vs_template(model, test_loader, template_loader, device):
 
 
 if __name__ == "__main__":
-    dataset_root = '../../../datasets/classification/LFWPairs/lfw-py/lfw_test_template_50_cropped'
-    test_pth = "../../../parameters/classification/swin_face/swin_face.pth"
-    test_pth_infected = "../../../parametersProcess/swin_face/swin_evilfiles_16.pth"
-    test_pth_flipped = "../../../parametersProcess/swin_face/swin_flip_16.pth"
+    dataset_root = '../datasets/LFWPairs/lfw-py/lfw_test_template_50_cropped'
+    test_pth = "../checkpoints/swin_face.pth"
+    test_pth_infected = "../parametersProcess/swin_face/swin_evilfiles_16.pth"
+    test_pth_flipped = "../parametersProcess/swin_face/swin_flip_16.pth"
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
     transform = transforms.Compose([
@@ -196,7 +196,7 @@ if __name__ == "__main__":
     
     print(model) # 输出模型结构
     
-    model.load_state_dict(torch.load(test_pth_flipped, map_location=device))
+    model.load_state_dict(torch.load(test_pth, map_location=device))
     model.to(device)
 
     evaluate_test_vs_template(model, test_loader, template_loader, device)