Skip to content

Commit

Permalink
.imgtools directory
Browse files Browse the repository at this point in the history
  • Loading branch information
fishingguy456 committed May 24, 2022
1 parent 9a8ff65 commit 455c027
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions imgtools/ops/ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ def __init__(self,
####### CRAWLER ############
# Checks if dataset has already been indexed
# To be changed later
path_crawl = os.path.join(self.parent, f"imgtools_{self.dataset_name}.csv")
path_crawl = os.path.join(self.parent, ".imgtools", f"imgtools_{self.dataset_name}.csv")
if not os.path.exists(path_crawl):
print("Couldn't find the dataset index CSV. Indexing the dataset...")
db = crawl(self.dir_path, n_jobs=n_jobs)
Expand All @@ -94,7 +94,7 @@ def __init__(self,

####### GRAPH ##########
# Form the graph
edge_path = os.path.join(self.parent,f"imgtools_{self.dataset_name}_edges.csv")
edge_path = os.path.join(self.parent,".imgtools",f"imgtools_{self.dataset_name}_edges.csv")
graph = DataGraph(path_crawl=path_crawl, edge_path=edge_path, visualize=visualize)
print(f"Forming the graph based on the given modalities: {self.modalities}")
self.df_combined = graph.parser(self.modalities)
Expand Down
2 changes: 1 addition & 1 deletion imgtools/utils/crawl.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def crawl(top,
# save one level above imaging folders
parent, dataset = os.path.split(top)

parent_imgtools = os.path.join(parent, "imgtools")
parent_imgtools = os.path.join(parent, ".imgtools")

if not os.path.exists(parent_imgtools):
os.makedirs(parent_imgtools)
Expand Down

0 comments on commit 455c027

Please sign in to comment.