avoid load colorfile many times

CSAILVision · Oct 23, 2018 · 276994b · 276994b
1 parent 0539224
commit 276994b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/eval.py b/eval.py
@@ -17,9 +17,10 @@
 import lib.utils.data as torchdata
 import cv2
 
+colors = loadmat('data/color150.mat')['colors']
+
 
 def visualize_result(data, preds, args):
-    colors = loadmat('data/color150.mat')['colors']
     (img, seg, info) = data
 
     # segmentation

diff --git a/eval_multipro.py b/eval_multipro.py
@@ -20,9 +20,10 @@
 import cv2
 from tqdm import tqdm
 
+colors = loadmat('data/color150.mat')['colors']
+
 
 def visualize_result(data, preds, args):
-    colors = loadmat('data/color150.mat')['colors']
     (img, seg, info) = data
 
     # segmentation