Skip to content

Commit

Permalink
moved graph template dir to utils
Browse files Browse the repository at this point in the history
  • Loading branch information
franioli committed Mar 30, 2024
1 parent 265cbc8 commit 597a942
Show file tree
Hide file tree
Showing 12 changed files with 3 additions and 2 deletions.
Empty file.
5 changes: 3 additions & 2 deletions src/deep_image_matching/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@

logger = logging.getLogger("dim")

TEMPLATE_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "utils/templates")


def view_graph(db, output_dir, imgs_dir):
logger.info("Creating view graph visualization...")
Expand All @@ -23,8 +25,7 @@ def view_graph(db, output_dir, imgs_dir):
nt = Network()

# HTML template for view graph details panel
current_directory = os.path.dirname(os.path.realpath(__file__))
nt.set_template(os.path.join(current_directory, "templates", "template.html"))
nt.set_template(os.path.join(TEMPLATE_DIR, "template.html"))

# Add nodes
G = nx.Graph()
Expand Down

0 comments on commit 597a942

Please sign in to comment.