Rename cbir_system.py to main.py

cbir_system.py → main.py

@@ -22,18 +22,18 @@ with open('list_eval_partition.txt', 'r') as f:
 # For a model pretrained on VGGFace2
 print('Loading model weights ........')
 
-# class SiameseModel(nn.Module):
-#     def __init__(self):
-#         super().__init__()
-#         self.backbone = InceptionResnetV1(pretrained='vggface2')
-#     def forward(self, x):
-#         x = self.backbone(x)
-#         x = torch.nn.functional.normalize(x, dim=1)
-#         return x
+class SiameseModel(nn.Module):
+    def __init__(self):
+        super().__init__()
+        self.backbone = InceptionResnetV1(pretrained='vggface2')
+    def forward(self, x):
+        x = self.backbone(x)
+        x = torch.nn.functional.normalize(x, dim=1)
+        return x
     
-# model = SiameseModel()
-# model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
-# model.eval()
+model = SiameseModel()
+model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
+model.eval()
 
 
 # Make FAISS index