(self, image_path)
| 88 | return mask |
| 89 | |
| 90 | def get_predicted_depth_path(self, image_path): |
| 91 | pred_dir = image_path.replace("JPEGImages", "depthImages") |
| 92 | pred_dir = osp.dirname(pred_dir) |
| 93 | os.makedirs(pred_dir, exist_ok=True) |
| 94 | base_name = osp.splitext(osp.basename(image_path))[0] + "_pred.png" |
| 95 | return osp.join(pred_dir, base_name) |
| 96 | |
| 97 | def __len__(self): |
| 98 | return len(self.ret_frames) |