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.
|
| 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/
|
| 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()))
|