(self, image_path)
| 94 | ]) |
| 95 | |
| 96 | def get_predicted_depth_path(self, image_path): |
| 97 | pred_dir = image_path.replace("JPEGImages", "depthImages") |
| 98 | pred_dir = osp.dirname(pred_dir) |
| 99 | os.makedirs(pred_dir, exist_ok=True) |
| 100 | base_name = osp.splitext(osp.basename(image_path))[0] + "_pred.png" |
| 101 | return osp.join(pred_dir, base_name) |
| 102 | |
| 103 | def __len__(self): |
| 104 | return len(self.all_frames) |