Tests

madhavkhoslaa · Jul 23, 2019 · 4463bb1 · 4463bb1
1 parent 9b9f88d
commit 4463bb1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test.py b/test/test.py
@@ -1,9 +1,11 @@
+import os
+os.chdir("..")
 from utils.one_hot_encoder import HotEncoder
 import  numpy as np
 import torch
 import skimage
 from torch.nn.functional import one_hot
-img__= skimage.io.imread("/Users/madhav/DataSets/data_semantics/training/semantic_rgb/000035_10.tif")
+img__= skimage.io.imread("/Users/madhav/DataSets/data_semantics/training/semantic_rgb/000034_10.tif")
 encoder= HotEncoder(dir= '/Users/madhav/DataSets/data_semantics/training/semantic_rgb', extension="tif", is_binary= False, n_classes= 30)
 color_dict= encoder.gen_colors()
 onehottt= encoder.PerPixelClassMatrix(img__)