new model

fasterrcnn_gtr_sar_new.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c032f612b8dc0d1f894430d7ab98e9cc7ff0a529013523c621b9353fd47654b
+size 165853984

model.py

@@ -2,7 +2,7 @@ import torch
 import torchvision
 from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
 
-def load_model(model_path="fasterrcnn_gtr_sar.pth", num_classes=8): # 7 aircraft classes + 1 background
+def load_model(model_path="fasterrcnn_gtr_sar_new.pth", num_classes=8): # 7 aircraft classes + 1 background
     model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights=None)
     in_features = model.roi_heads.box_predictor.cls_score.in_features
     model.roi_heads.box_predictor = FastRCNNPredictor(in_features, num_classes)