diff --git a/Ch/0.jpeg b/Ch/0.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/0.jpeg differ diff --git a/Ch/000035_10 copy.tif b/Ch/000035_10 copy.tif new file mode 100644 index 0000000..7ebf409 Binary files /dev/null and b/Ch/000035_10 copy.tif differ diff --git a/Ch/1.jpeg b/Ch/1.jpeg new file mode 100644 index 0000000..8c9cc77 Binary files /dev/null and b/Ch/1.jpeg differ diff --git a/Ch/10.jpeg b/Ch/10.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/10.jpeg differ diff --git a/Ch/11.jpeg b/Ch/11.jpeg new file mode 100644 index 0000000..fe96d3d Binary files /dev/null and b/Ch/11.jpeg differ diff --git a/Ch/12.jpeg b/Ch/12.jpeg new file mode 100644 index 0000000..a539784 Binary files /dev/null and b/Ch/12.jpeg differ diff --git a/Ch/13.jpeg b/Ch/13.jpeg new file mode 100644 index 0000000..d28e4f3 Binary files /dev/null and b/Ch/13.jpeg differ diff --git a/Ch/14.jpeg b/Ch/14.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/14.jpeg differ diff --git a/Ch/15.jpeg b/Ch/15.jpeg new file mode 100644 index 0000000..9632a60 Binary files /dev/null and b/Ch/15.jpeg differ diff --git a/Ch/16.jpeg b/Ch/16.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/16.jpeg differ diff --git a/Ch/17.jpeg b/Ch/17.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/17.jpeg differ diff --git a/Ch/18.jpeg b/Ch/18.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/18.jpeg differ diff --git a/Ch/19.jpeg b/Ch/19.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/19.jpeg differ diff --git a/Ch/2.jpeg b/Ch/2.jpeg new file mode 100644 index 0000000..2c19d3a Binary files /dev/null and b/Ch/2.jpeg differ diff --git a/Ch/20.jpeg b/Ch/20.jpeg new file mode 100644 index 0000000..d05f189 Binary files /dev/null and b/Ch/20.jpeg differ diff --git a/Ch/21.jpeg b/Ch/21.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/21.jpeg differ diff --git a/Ch/22.jpeg b/Ch/22.jpeg new file mode 100644 index 0000000..5a5e99c Binary files /dev/null and b/Ch/22.jpeg differ diff --git a/Ch/23.jpeg b/Ch/23.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/23.jpeg differ diff --git a/Ch/24.jpeg b/Ch/24.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/24.jpeg differ diff --git a/Ch/25.jpeg b/Ch/25.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/25.jpeg differ diff --git a/Ch/26.jpeg b/Ch/26.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/26.jpeg differ diff --git a/Ch/27.jpeg b/Ch/27.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/27.jpeg differ diff --git a/Ch/28.jpeg b/Ch/28.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/28.jpeg differ diff --git a/Ch/29.jpeg b/Ch/29.jpeg new file mode 100644 index 0000000..08bafdf Binary files /dev/null and b/Ch/29.jpeg differ diff --git a/Ch/3.jpeg b/Ch/3.jpeg new file mode 100644 index 0000000..76c5679 Binary files /dev/null and b/Ch/3.jpeg differ diff --git a/Ch/4.jpeg b/Ch/4.jpeg new file mode 100644 index 0000000..07c358a Binary files /dev/null and b/Ch/4.jpeg differ diff --git a/Ch/5.jpeg b/Ch/5.jpeg new file mode 100644 index 0000000..3c4d8f9 Binary files /dev/null and b/Ch/5.jpeg differ diff --git a/Ch/6.jpeg b/Ch/6.jpeg new file mode 100644 index 0000000..e0bd9f6 Binary files /dev/null and b/Ch/6.jpeg differ diff --git a/Ch/7.jpeg b/Ch/7.jpeg new file mode 100644 index 0000000..64c7483 Binary files /dev/null and b/Ch/7.jpeg differ diff --git a/Ch/8.jpeg b/Ch/8.jpeg new file mode 100644 index 0000000..5006de0 Binary files /dev/null and b/Ch/8.jpeg differ diff --git a/Ch/9.jpeg b/Ch/9.jpeg new file mode 100644 index 0000000..261780f Binary files /dev/null and b/Ch/9.jpeg differ diff --git a/test.py b/test.py index 17ec339..48a5de8 100644 --- a/test.py +++ b/test.py @@ -1,24 +1,10 @@ from utils.one_hot_encoder import HotEncoder import numpy as np +import torch import skimage -img__= skimage.io.imread("/Users/madhav/Desktop/Screenshot 2019-07-17 at 11.27.51 PM.png") -encoder= HotEncoder(dir= "/Users/madhav/Desktop/", extension="png", is_binary= False) +from torch.nn.functional import one_hot +img__= skimage.io.imread("/Users/madhav/DataSets/data_semantics/training/semantic_rgb/000035_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() - -def rgb_to_onehot(rgb_arr, color_dict): - num_classes = len(color_dict) - shape = rgb_arr.shape[:2]+(num_classes,) - arr = np.zeros( shape, dtype=np.int8 ) - for i, cls in enumerate(color_dict): - arr[:,:,i] = np.all(rgb_arr.reshape( (-1,3) ) == color_dict[i], axis=1).reshape(shape[:2]) - return arr - - -def onehot_to_rgb(onehot, color_dict): - single_layer = np.argmax(onehot, axis=-1) - output = np.zeros( onehot.shape[:2]+(3,) ) - for k in color_dict.keys(): - output[single_layer==k] = color_dict[k] - return np.uint8(output) -class_mat= encoder.PerPixelClassMatrix(img__) -print(class_mat) +onehottt= encoder.PerPixelClassMatrix(img__) +encoded= encoder.HotEncode(onehottt)