run on cpu

app.py

@@ -45,6 +45,9 @@ article = """
 <p style='text-align: center'><a href='https://arxiv.org/abs/2212.03239' target='_blank'>Perspective Fields for Single Image Camera Calibrations</a> | <a href='https://github.com/jinlinyi/PerspectiveFields' target='_blank'>Github Repo</a></p>
 """
 
+
+
+
 def setup_cfg(args):
     cfgs = []
     configs = args['config_file'].split('#')
@@ -146,23 +149,24 @@ for img_name in glob('assets/imgs/*.*g'):
     examples.append([img_name])
 print(examples)
 
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model_zoo = {
     'PersNet-360Cities': {
         'weights': ['https://www.dropbox.com/s/czqrepqe7x70b7y/cvpr2023.pth'],
-        'opts': ['MODEL.WEIGHTS', 'models/cvpr2023.pth'],
+        'opts': ['MODEL.WEIGHTS', 'models/cvpr2023.pth', 'MODEL.DEVICE', device,],
         'config_file': 'models/cvpr2023.yaml',
         'param': False,
     },
     'PersNet_Paramnet-GSV-uncentered': {
         'weights': ['https://www.dropbox.com/s/ufdadxigewakzlz/paramnet_gsv_rpfpp.pth'],
-        'opts': ['MODEL.WEIGHTS', 'models/paramnet_gsv_rpfpp.pth'],
+        'opts': ['MODEL.WEIGHTS', 'models/paramnet_gsv_rpfpp.pth', 'MODEL.DEVICE', device,],
         'config_file': 'models/paramnet_gsv_rpfpp.yaml',
         'param': True,
     },
     # trained on GSV dataset, predicts Perspective Fields + camera parameters (roll, pitch, fov), assuming centered principal point
     'PersNet_Paramnet-GSV-centered': {
         'weights': ['https://www.dropbox.com/s/g6xwbgnkggapyeu/paramnet_gsv_rpf.pth'],
-        'opts': ['MODEL.WEIGHTS', 'models/paramnet_gsv_rpf.pth'],
+        'opts': ['MODEL.WEIGHTS', 'models/paramnet_gsv_rpf.pth', 'MODEL.DEVICE', device,],
         'config_file': 'models/paramnet_gsv_rpf.yaml',
         'param': True,
     },