>

fregu856 · Sep 27, 2018 · 8126495 · 8126495
1 parent 128536b
commit 8126495
Show file tree

Hide file tree

Showing 12 changed files with 35 additions and 6 deletions.
diff --git a/datasets.py b/datasets.py
@@ -1,4 +1,4 @@
-# camera-ready
+# camera-ready if everything works
 
 import torch
 import torch.utils.data

diff --git a/eval_on_val.py → evaluation/eval_on_val.py b/eval_on_val.py → evaluation/eval_on_val.py
@@ -1,8 +1,14 @@
-# camera-ready
+# camera-ready if everything works
 
+import sys
+
+sys.path.append("/root/deeplabv3")
 from datasets import DatasetVal # (this needs to be imported before torch, because cv2 needs to be imported before torch for some reason)
+
+sys.path.append("/root/deeplabv3/model")
 from deeplabv3 import DeepLabV3
 
+sys.path.append("/root/deeplabv3/utils")
 from utils import label_img_to_color
 
 import torch

diff --git a/eval_on_val_for_metrics.py → evaluation/eval_on_val_for_metrics.py b/eval_on_val_for_metrics.py → evaluation/eval_on_val_for_metrics.py
@@ -1,8 +1,14 @@
-# camera-ready
+# camera-ready if everything works
 
+import sys
+
+sys.path.append("/root/deeplabv3")
 from datasets import DatasetVal # (this needs to be imported before torch, because cv2 needs to be imported before torch for some reason)
+
+sys.path.append("/root/deeplabv3/model")
 from deeplabv3 import DeepLabV3
 
+sys.path.append("/root/deeplabv3/utils")
 from utils import label_img_to_color
 
 import torch

diff --git a/aspp.py → model/aspp.py b/aspp.py → model/aspp.py
diff --git a/deeplabv3.py → model/deeplabv3.py b/deeplabv3.py → model/deeplabv3.py
diff --git a/resnet.py → model/resnet.py b/resnet.py → model/resnet.py
diff --git a/train.py b/train.py
@@ -1,8 +1,13 @@
-# camera-ready
+# camera-ready if everything works
+
+import sys
 
 from datasets import DatasetTrain, DatasetVal # (this needs to be imported before torch, because cv2 needs to be imported before torch for some reason)
+
+sys.path.append("/root/deeplabv3/model")
 from deeplabv3 import DeepLabV3
 
+sys.path.append("/root/deeplabv3/utils")
 from utils import add_weight_decay
 
 import torch

diff --git a/preprocess_data.py → utils/preprocess_data.py b/preprocess_data.py → utils/preprocess_data.py
diff --git a/random_code.py → utils/random_code.py b/random_code.py → utils/random_code.py
diff --git a/utils.py → utils/utils.py b/utils.py → utils/utils.py
diff --git a/run_on_seq.py → visualization/run_on_seq.py b/run_on_seq.py → visualization/run_on_seq.py
@@ -1,8 +1,14 @@
-# camera-ready
+# camera-ready if everything works
 
+import sys
+
+sys.path.append("/root/deeplabv3")
 from datasets import DatasetSeq # (this needs to be imported before torch, because cv2 needs to be imported before torch for some reason)
+
+sys.path.append("/root/deeplabv3/model")
 from deeplabv3 import DeepLabV3
 
+sys.path.append("/root/deeplabv3/utils")
 from utils import label_img_to_color
 
 import torch

diff --git a/run_on_thn_seq.py → visualization/run_on_thn_seq.py b/run_on_thn_seq.py → visualization/run_on_thn_seq.py
@@ -1,8 +1,14 @@
-# camera-ready
+# camera-ready if everything works
 
+import sys
+
+sys.path.append("/root/deeplabv3")
 from datasets import DatasetThnSeq # (this needs to be imported before torch, because cv2 needs to be imported before torch for some reason)
+
+sys.path.append("/root/deeplabv3/model")
 from deeplabv3 import DeepLabV3
 
+sys.path.append("/root/deeplabv3/utils")
 from utils import label_img_to_color
 
 import torch