Fixed default segmentation output shapes

alexsax · Jul 18, 2019 · f357e1e · f357e1e
1 parent 455b0f2
commit f357e1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/visualpriors/taskonomy_network.py b/visualpriors/taskonomy_network.py
@@ -37,9 +37,9 @@
     1: ['colorization', 'edge_texture', 'edge_occlusion',  'keypoints3d', 'keypoints2d', 'reshading', 'depth_zbuffer', 'depth_euclidean', ],
     2: ['curvature', 'principal_curvature'],
     3: ['autoencoding', 'denoising', 'normal', 'inpainting', 'rgb', 'normals'],
-    128: ['segment_unsup2d', 'segment_unsup25d'],
+    17: ['segment_semantic'],
+    64: ['segment_unsup2d', 'segment_unsup25d'],
     1000: ['class_object'],
-    None: ['segment_semantic']
 }