try install offline
Browse files- app.py +5 -5
- 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 |
+
|