Update codes.py

codes.py

@@ -62,18 +62,6 @@ class Build_IRmodel_Resnet(nn.Module):
             predction = {'Affine_mtrx': Prd_Affine_mtrx}
         return predction
 
-from torchvision.models import resnet18
-
-core_model_tst = resnet18(pretrained=True)
-core_model_tst.fc = Identity()
-core_model_tst.load_state_dict(torch.load(file_savingfolder+'core_model'+ext+'.pth'))
-core_model_tst.to(device)
-IR_Model_tst = Build_IRmodel_Resnet(core_model_tst, registration_method)
-IR_Model_tst.load_state_dict(torch.load(file_savingfolder+'IR_Model'+ext+'.pth'))
-IR_Model_tst.to(device)
-IR_Model_tst.eval()
-
-
 def pil_to_numpy(im):
     im.load()
     # Unpack data