(self, index)
| 39 | return gt_pose |
| 40 | |
| 41 | def load_depth(self, index): |
| 42 | depth = cv2.imread( |
| 43 | osp.join(self.data_path, 'results/depth{:06d}.png'.format(index)), -1) |
| 44 | depth = depth / 6553.5 |
| 45 | if self.max_depth > 0: |
| 46 | depth[depth > self.max_depth] = 0 |
| 47 | return depth |
| 48 | |
| 49 | def load_image(self, index): |
| 50 | rgb = cv2.imread( |