jinlinyi commited on
Commit
fb147c0
·
1 Parent(s): f73740f

try install offline

Browse files
Files changed (2) hide show
  1. app.py +5 -5
  2. requirements.txt +4 -0
app.py CHANGED
@@ -1,8 +1,8 @@
1
 
2
  import os
3
- os.system(f"pip install -U openmim")
4
- os.system(f"mim install mmcv")
5
- os.system(f"pip install git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d")
6
 
7
 
8
 
@@ -141,14 +141,14 @@ print(examples)
141
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
142
  model_zoo = {
143
 
144
- 'Paramnet-360Cities-edina-centered': {
145
  'weights': ['https://www.dropbox.com/s/z2dja70bgy007su/paramnet_360cities_edina_rpf.pth'],
146
  'opts': ['MODEL.WEIGHTS', 'models/paramnet_360cities_edina_rpf.pth', 'MODEL.DEVICE', device,],
147
  'config_file': 'models/paramnet_360cities_edina_rpf.yaml',
148
  'param': True,
149
  },
150
 
151
- 'Paramnet-360Cities-edina-uncentered': {
152
  'weights': ['https://www.dropbox.com/s/nt29e1pi83mm1va/paramnet_360cities_edina_rpfpp.pth'],
153
  'opts': ['MODEL.WEIGHTS', 'models/paramnet_360cities_edina_rpfpp.pth', 'MODEL.DEVICE', device,],
154
  'config_file': 'models/paramnet_360cities_edina_rpfpp.yaml',
 
1
 
2
  import os
3
+ # os.system(f"pip install -U openmim")
4
+ # os.system(f"mim install mmcv")
5
+ # os.system(f"pip install git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d")
6
 
7
 
8
 
 
141
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
142
  model_zoo = {
143
 
144
+ 'NEW:Paramnet-360Cities-edina-centered': {
145
  'weights': ['https://www.dropbox.com/s/z2dja70bgy007su/paramnet_360cities_edina_rpf.pth'],
146
  'opts': ['MODEL.WEIGHTS', 'models/paramnet_360cities_edina_rpf.pth', 'MODEL.DEVICE', device,],
147
  'config_file': 'models/paramnet_360cities_edina_rpf.yaml',
148
  'param': True,
149
  },
150
 
151
+ 'NEW:Paramnet-360Cities-edina-uncentered': {
152
  'weights': ['https://www.dropbox.com/s/nt29e1pi83mm1va/paramnet_360cities_edina_rpfpp.pth'],
153
  'opts': ['MODEL.WEIGHTS', 'models/paramnet_360cities_edina_rpfpp.pth', 'MODEL.DEVICE', device,],
154
  'config_file': 'models/paramnet_360cities_edina_rpfpp.yaml',
requirements.txt CHANGED
@@ -1,3 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch==1.11.0+cu113
3
  torchvision==0.12.0+cu113
 
 
 
 
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch==1.11.0+cu113
3
  torchvision==0.12.0+cu113
4
+ --find-links https://download.openmmlab.com/mmcv/dist/cu113/torch1.11/index.html
5
+ mmcv==2.0.0rc4
6
+ git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d
7
+