UPDATE MODEL

model.py

@@ -8,7 +8,7 @@ from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
 def create_model(num_classes):
     
     # load Faster RCNN pre-trained model
-    model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights = "DEFAULT" ,pretrained=True, pretrained_backbone=True, progress=True, trainable_backbone_layers=4)
+    model = torchvision.models.detection.fasterrcnn_resnet50_fpn_v2(weights = "COCO_V1" ,pretrained=True, weights_backbone = 'IMAGENET1K_V1', progress=True)
     
     # get the number of input features 
     in_features = model.roi_heads.box_predictor.cls_score.in_features