Spaces:
Runtime error
Runtime error
fixed path
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
import os
|
| 3 |
import sys
|
| 4 |
os.system("git clone https://github.com/NVlabs/GroupViT")
|
| 5 |
-
|
| 6 |
|
| 7 |
import os.path as osp
|
| 8 |
from collections import namedtuple
|
|
@@ -23,6 +23,7 @@ from segmentation.evaluation import (GROUP_PALETTE, build_seg_demo_pipeline,
|
|
| 23 |
build_seg_inference)
|
| 24 |
from utils import get_config, load_checkpoint
|
| 25 |
|
|
|
|
| 26 |
checkpoint_url = 'https://github.com/xvjiarui/GroupViT-1/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
|
| 27 |
cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
|
| 28 |
output_dir = 'demo/output'
|
|
|
|
| 2 |
import os
|
| 3 |
import sys
|
| 4 |
os.system("git clone https://github.com/NVlabs/GroupViT")
|
| 5 |
+
sys.path.insert(0, 'GroupViT')
|
| 6 |
|
| 7 |
import os.path as osp
|
| 8 |
from collections import namedtuple
|
|
|
|
| 23 |
build_seg_inference)
|
| 24 |
from utils import get_config, load_checkpoint
|
| 25 |
|
| 26 |
+
os.chdir('GroupViT')
|
| 27 |
checkpoint_url = 'https://github.com/xvjiarui/GroupViT-1/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
|
| 28 |
cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
|
| 29 |
output_dir = 'demo/output'
|