minor update

bertjiazheng · Mar 25, 2020 · f30d630 · f30d630
1 parent c90b689
commit f30d630
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/visualize_bbox.py b/visualize_bbox.py
@@ -29,9 +29,10 @@ def visualize_bbox(args):
             position_path = os.path.join(room_path, position_id)
 
             image = cv2.imread(os.path.join(position_path, 'rgb_rawlight.png'))
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
             height, width, _ = image.shape
 
-            instance = cv2.imread(os.path.join(position_path, 'instance.png'), -1)
+            instance = cv2.imread(os.path.join(position_path, 'instance.png'), cv2.IMREAD_UNCHANGED)
 
             camera_info = np.loadtxt(os.path.join(position_path, 'camera_pose.txt'))
 

diff --git a/visualize_layout.py b/visualize_layout.py
@@ -22,6 +22,7 @@ def visualize_panorama(args):
 
         cor_id = np.loadtxt(os.path.join(room_path, "layout.txt"))
         img_src = cv2.imread(os.path.join(room_path, "full", "rgb_rawlight.png"))
+        img_src = cv2.cvtColor(img_src, cv2.COLOR_BGR2RGB)
         img_viz = draw_boundary_from_cor_id(cor_id, img_src)
 
         plt.axis('off')
@@ -46,6 +47,7 @@ def visualize_perspective(args):
             position_path = os.path.join(room_path, position_id)
 
             image = cv2.imread(os.path.join(position_path, "rgb_rawlight.png"))
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
             with open(os.path.join(position_path, "layout.json")) as f:
                 annos = json.load(f)