changed cnn architecture

bird_cnn/server.py

@@ -12,13 +12,7 @@ import torch.nn.functional as F
 from bird_cnn import Bird_CNN
 
 BUILD_PATH = "./build_models"
-#IMAGE_SIZE = (1141, 850)
-IMAGE_SIZE = (300, 300)
-
-transform = transforms.Compose([
-    transforms.Resize(IMAGE_SIZE),
-    transforms.ToTensor()
-])
+IMAGE_SIZE = 256
 
 class bird_species(Enum):
     Common_Kingfisher = 0
@@ -31,6 +25,7 @@ class bird_species(Enum):
 
 transform = transforms.Compose([
     transforms.Resize(IMAGE_SIZE),
+    transforms.CenterCrop(IMAGE_SIZE),
     transforms.ToTensor()
 ])