Skip to content

GitLab

Unverified Commit 3e2adb5d authored by Vitalii Kudinov's avatar Vitalii Kudinov Committed by GitHub
Browse files

Fix for SUN RGB-D, when creating .pkl files (#511)

parent 59164170
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/data_converter/sunrgbd_data_utils.py
View file @ 3e2adb5d
...@@ -113,7 +113,7 @@ class SUNRGBDData(object): ...@@ -113,7 +113,7 @@ class SUNRGBDData(object):
Rt = np.array([float(x) for x in lines[0].split(' ')]) Rt = np.array([float(x) for x in lines[0].split(' ')])
Rt = np.reshape(Rt, (3, 3), order='F').astype(np.float32) Rt = np.reshape(Rt, (3, 3), order='F').astype(np.float32)
K = np.array([float(x) for x in lines[1].split(' ')]) K = np.array([float(x) for x in lines[1].split(' ')])
K = np.reshape(Rt, (3, 3), order='F').astype(np.float32) K = np.reshape(K, (3, 3), order='F').astype(np.float32)
return K, Rt return K, Rt
def get_label_objects(self, idx): def get_label_objects(self, idx):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment