update model path

rcnn_model/scripts/rcnn_run.py

@@ -26,6 +26,7 @@ from rcnn_model.extraction.annotation_builder import AnnotationBuilder as AnnBui
 
 results_directory = "/tmp/results/"
 sample_data_directory = "rcnn_model/sample/"
+model_directory = "tmp/output/"
 
 def main(cfg,img_source_path, coco_dest_filename, val_img_dest_filename):
     os.chdir(str(from_root()))
@@ -33,8 +34,9 @@ def main(cfg,img_source_path, coco_dest_filename, val_img_dest_filename):
     print("Inside main: " + img_source_path)
     #configure model
     cfg.DATALOADER.NUM_WORKERS = 1
-    cfg.SOLVER.IMS_PER_BATCH = 1                                           
-    cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
+    cfg.SOLVER.IMS_PER_BATCH = 1 
+    # Changed this file here                                          
+    cfg.MODEL.WEIGHTS = os.path.join(model_directory, "model_final.pth")
     cfg.MODEL.DEVICE = "cpu"
     predictor = DefaultPredictor(cfg)