Dharini Baskaran commited on
Commit
bf8c65c
·
1 Parent(s): 1f3b592

updated the model weight path

Browse files
rcnn_model/scripts/rcnn_eval.py CHANGED
@@ -23,7 +23,7 @@ def main(cfg,results_filename = "eval_results.txt"):
23
  #update config file
24
  cfg.DATALOADER.NUM_WORKERS = 1
25
  cfg.SOLVER.IMS_PER_BATCH = 1
26
- cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final_2.pth")
27
 
28
  #run evaluation
29
  results = standard_evaluation(cfg)
 
23
  #update config file
24
  cfg.DATALOADER.NUM_WORKERS = 1
25
  cfg.SOLVER.IMS_PER_BATCH = 1
26
+ cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR2, "model_final_2.pth")
27
 
28
  #run evaluation
29
  results = standard_evaluation(cfg)
rcnn_model/scripts/rcnn_run.py CHANGED
@@ -13,7 +13,7 @@ from rcnn_model.extraction.annotation_builder import AnnotationBuilder as AnnBui
13
 
14
  results_directory = "/tmp/results/"
15
  sample_data_directory = "rcnn_model/sample/"
16
- model_directory = "/tmp/output/"
17
 
18
  def main(cfg,img_source_path, coco_dest_filename, val_img_dest_filename):
19
  os.chdir(str(from_root()))
 
13
 
14
  results_directory = "/tmp/results/"
15
  sample_data_directory = "rcnn_model/sample/"
16
+ model_directory = "/tmp/output2/"
17
 
18
  def main(cfg,img_source_path, coco_dest_filename, val_img_dest_filename):
19
  os.chdir(str(from_root()))