hujiecpp commited on
Commit
1fa3bc9
·
1 Parent(s): 1222c66

init project

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  2. requirements.txt +9 -1
app.py CHANGED
@@ -8,7 +8,10 @@ import os
8
  import sys
9
  sys.path.append(os.path.abspath('./modules'))
10
 
11
- sys.path.append('naver/mast3r/')
 
 
 
12
 
13
  import math
14
  import tempfile
@@ -45,7 +48,6 @@ sys.path.append(os.path.abspath('./modules/ultralytics'))
45
 
46
  from transformers import AutoTokenizer, AutoModel, AutoProcessor, SamModel
47
  # from modules.mast3r.model import AsymmetricMASt3R
48
- from mast3r.model import AsymmetricMASt3R
49
 
50
  # from modules.sam2.build_sam import build_sam2_video_predictor
51
  from modules.mobilesamv2.promt_mobilesamv2 import ObjectAwareModel
 
8
  import sys
9
  sys.path.append(os.path.abspath('./modules'))
10
 
11
+ HERE_PATH = os.path.normpath(os.path.dirname(__file__)) # noqa
12
+ MASt3R_REPO_PATH = os.path.normpath(os.path.join(HERE_PATH, './mast3r')) # noqa
13
+ sys.path.insert(0, MASt3R_REPO_PATH) # noqa
14
+ from mast3r.model import AsymmetricMASt3R
15
 
16
  import math
17
  import tempfile
 
48
 
49
  from transformers import AutoTokenizer, AutoModel, AutoProcessor, SamModel
50
  # from modules.mast3r.model import AsymmetricMASt3R
 
51
 
52
  # from modules.sam2.build_sam import build_sam2_video_predictor
53
  from modules.mobilesamv2.promt_mobilesamv2 import ObjectAwareModel
requirements.txt CHANGED
@@ -10,4 +10,12 @@ matplotlib
10
  scipy
11
  roma
12
  hydra-core
13
- iopath
 
 
 
 
 
 
 
 
 
10
  scipy
11
  roma
12
  hydra-core
13
+ iopath
14
+ numpy<2.0.0
15
+ scikit-learn
16
+ pillow-heif
17
+ gradio
18
+ tqdm
19
+ tensorboard
20
+ pyglet<2
21
+ huggingface-hub[torch]>=0.22