fix: minor bug

UTAustin-SwarmLab · Sep 2, 2023 · 4982d09 · 4982d09
1 parent 8a7c078
commit 4982d09
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/main.py b/main.py
@@ -1,11 +1,11 @@
 from __future__ import annotations
 
-from video_to_automaton.config.loader import load_config
-from video_to_automaton.model.vision.grounding_dino import GroundingDino
-from video_to_automaton.processor.video_processor import (
+from ns_vfs.config.loader import load_config
+from ns_vfs.model.vision.grounding_dino import GroundingDino
+from ns_vfs.processor.video_processor import (
     VideoFrameProcessor,
 )
-from video_to_automaton.video_to_automaton import VideotoAutomaton
+from ns_vfs.video_to_automaton import VideotoAutomaton
 
 if __name__ == "__main__":
     sample_video_path = "/opt/Video-to-Automoton/artifacts/data/hmdb51/clap/Applauding_Abby_clap_u_nm_np1_fr_med_1.avi"

diff --git a/ns_vfs/config/__init__.py b/ns_vfs/config/__init__.py
@@ -2,9 +2,7 @@
 
 from ns_vfs.common import omegaconf
 
-ROOT_DIR = os.path.dirname(os.path.abspath(__file__)).split(
-    "video_to_automaton"
-)[0]
+ROOT_DIR = os.path.dirname(os.path.abspath(__file__)).split("ns_vfs")[0]
 
 config_path = os.path.join(
     os.path.dirname(os.path.abspath(__file__)), "config.yaml"

diff --git a/ns_vfs/config/loader.py b/ns_vfs/config/loader.py
@@ -22,7 +22,7 @@ def load_config():
     )
     config.GROUNDING_DINO.GROUNDING_DINO_CONFIG_PATH = os.path.join(
         config.VERSION_AND_PATH.ROOT_PATH,
-        "video_to_automaton",
+        "ns_vfs",
         "config",
         "GroundingDINO_SwinT_OGC.py",
     )