bug fix

Jianxff · May 4, 2024 · ca895d3 · ca895d3
1 parent 8cebf1a
commit ca895d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/data.py b/modules/data.py
@@ -62,8 +62,8 @@ def __getitem__(self, idx) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
         raw_rgb = cv2.imread(str(self.rgb_list[idx]), cv2.IMREAD_COLOR).astype(np.uint8)
         raw_depth = None if not self.depth_list else np.load(self.depth_list[idx])
         # process data
-        rgb = cv2.resize(rgb, self.resize) if self.resize else raw_rgb
-        depth = None if not self.depth_list else (cv2.resize(depth, self.resize) if self.resize else raw_depth)
+        rgb = cv2.resize(raw_rgb, self.resize) if self.resize else raw_rgb
+        depth = None if not self.depth_list else (cv2.resize(raw_depth, self.resize) if self.resize else raw_depth)
         pose = None if not self.pose_list else np.loadtxt(self.pose_list[idx])
 
         return rgb, depth, pose, self.intrinsic