diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index 3f667d6fee5c5b9fb838e44c8e1b9508624d91b3..b887a81c60ef019a3eed778ebbaf5120f117f845 100755 --- a/datasets/linemod/dataset.py +++ b/datasets/linemod/dataset.py @@ -154,8 +154,7 @@ class PoseDataset(data.Dataset): pt0 = (ymap_masked - self.cam_cx) * pt2 / self.cam_fx pt1 = (xmap_masked - self.cam_cy) * pt2 / self.cam_fy cloud = np.concatenate((pt0, pt1, pt2), axis=1) - cloud = np.add(cloud, -1.0 * target_t) / 1000.0 - cloud = np.add(cloud, target_t / 1000.0) + cloud = cloud / 1000.0 if self.add_noise: cloud = np.add(cloud, add_t)