init project
Browse files- app.py +4 -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 |
-
|
|
|
|
|
|
|
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
|