mv functions to gtad_lib

frostinassiky · Apr 15, 2020 · 8bc1678 · 8bc1678
1 parent ba68778
commit 8bc1678
Show file tree

Hide file tree

Showing 9 changed files with 17 additions and 11 deletions.
diff --git a/.gitignore b/.gitignore
@@ -102,3 +102,7 @@ venv.bak/
 
 # mypy
 .mypy_cache/
+
+data/thumos_annotations/saved*
+data/thumos_feature/TSN*
+output/*
diff --git a/gtad_inference.py b/gtad_inference.py
@@ -1,12 +1,12 @@
 import os
-import torch.nn.parallel
-import numpy as np
 import math
-import opts
-from models import GTAD
+import numpy as np
 import pandas as pd
+import torch.nn.parallel
 
-from dataset import VideoDataSet
+from gtad_lib import opts
+from gtad_lib.models import GTAD
+from gtad_lib.dataset import VideoDataSet
 
 if __name__ == '__main__':
     opt = opts.parse_opt()

diff --git a/dataset.py → gtad_lib/dataset.py b/dataset.py → gtad_lib/dataset.py
diff --git a/loss_function.py → gtad_lib/loss_function.py b/loss_function.py → gtad_lib/loss_function.py
diff --git a/models.py → gtad_lib/models.py b/models.py → gtad_lib/models.py
diff --git a/opts.py → gtad_lib/opts.py b/opts.py → gtad_lib/opts.py
diff --git a/gtad_postprocess.py b/gtad_postprocess.py
@@ -1,11 +1,12 @@
 import sys
-import opts
 import numpy as np
 import pandas as pd
 import json
 import os
 from joblib import Parallel, delayed
 
+from gtad_lib import opts
+
 thumos_class = {
     7 : 'BaseballPitch',
     9 : 'BasketballDunk',

diff --git a/gtad_train.py b/gtad_train.py
@@ -1,13 +1,14 @@
-from dataset import VideoDataSet
-from loss_function import get_mask, subgraph_loss_func, node_loss_func
 import os
 import torch
 import torch.nn.parallel
 import torch.optim as optim
-import numpy as np
-import opts
-from models import GTAD
 from torch import autograd
+import numpy as np
+
+from gtad_lib import opts
+from gtad_lib.models import GTAD
+from gtad_lib.dataset import VideoDataSet
+from gtad_lib.loss_function import get_mask, subgraph_loss_func, node_loss_func
 
 
 ################## fix everything ##################

diff --git a/uNet_test.npy b/uNet_test.npy