updated the model weight path

README.md

@@ -31,11 +31,11 @@ A Gradio web app that allows you to upload 2D floorplan images and automatically
 
 3. **Download the pretrained model:**
 
-    - Download `model_final.pth` from [Google Drive here](https://drive.google.com/file/d/1yr64AOgaYZPTcQzG6cxG6lWBENHR9qjW/view?usp=sharing).
+    - Download `model_final_2.pth` from [Google Drive here](https://drive.google.com/file/d/1TJLKFI-sSQQ4JOqDgS6aqcw26IsadXGO/view?usp=sharing).
     - Place it inside:
 
       ```plaintext
-      floorplan-vectorizer/rcnn_model/output/model_final.pth
+      floorplan-vectorizer/rcnn_model/output/model_final_2.pth
       ```
 
 4. **Run the app:**

app.py

@@ -17,7 +17,7 @@ JSON_DIR = "/tmp/results/"
 OUTPUT_DIR = "/tmp/output/"
 MODEL_DIR = os.path.join(BASE_DIR, "rcnn_model", "scripts")
 logo_path = os.path.join(BASE_DIR, "public", "logo.png")
-model_path = os.path.join(OUTPUT_DIR, "model_final.pth")
+model_path = os.path.join(OUTPUT_DIR, "model_final_2.pth")
 
 # Google Drive model
 GOOGLE_DRIVE_FILE_ID = "1ptdr6d_jglk_78e4d5s1jUO04XWS1zo"

rcnn_model/scripts/rcnn_eval.py

@@ -23,7 +23,7 @@ def main(cfg,results_filename = "eval_results.txt"):
     #update config file
     cfg.DATALOADER.NUM_WORKERS = 1
     cfg.SOLVER.IMS_PER_BATCH = 1                                           
-    cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
+    cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final_2.pth")
 
     #run evaluation
     results = standard_evaluation(cfg)

rcnn_model/scripts/rcnn_run.py

@@ -22,7 +22,7 @@ def main(cfg,img_source_path, coco_dest_filename, val_img_dest_filename):
     #configure model
     cfg.DATALOADER.NUM_WORKERS = 1
     cfg.SOLVER.IMS_PER_BATCH = 1                                        
-    cfg.MODEL.WEIGHTS = os.path.join(model_directory, "model_final.pth")
+    cfg.MODEL.WEIGHTS = os.path.join(model_directory, "model_final_2.pth")
     cfg.MODEL.DEVICE = "cpu"
     predictor = DefaultPredictor(cfg)