Merge pull request #211 from bacetiner/master

Updated FacadeParser for better prediction performance
NHERI-SimCenter · Jun 17, 2024 · 6a528b6 · 6a528b6
2 parents 04d2a17 + 823bc32
commit 6a528b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/brails/modules/FacadeParser/FacadeParser.py b/brails/modules/FacadeParser/FacadeParser.py
@@ -196,7 +196,7 @@ def install_default_model(model_path='tmp/models/facadeParser.pth'):
 
                 if not os.path.isfile(model_path):
                     print('Loading default facade parser model file to tmp/models folder...')
-                    torch.hub.download_url_to_file('https://zenodo.org/record/5809365/files/facadeParser.pth',
+                    torch.hub.download_url_to_file('https://zenodo.org/record/10448047/files/facadeParserv2.pth',
                                                    model_path, progress=False)
                     print('Default facade parser model loaded')
                 else: 
@@ -578,4 +578,4 @@ def get_depth_map(depthfile, imsize, bndangles):
         # Unload the model from GPU
         if torch.cuda.is_available():
             del model
-            torch.cuda.empty_cache()
+            torch.cuda.empty_cache()