(self, mask)
| 82 | ]) |
| 83 | |
| 84 | def transform_mask(self, mask): |
| 85 | mask = self.mask_transform(mask) |
| 86 | mask = np.array(mask, dtype=np.uint8) |
| 87 | mask = torch.from_numpy(mask) |
| 88 | return mask |
| 89 | |
| 90 | def get_predicted_depth_path(self, image_path): |
| 91 | pred_dir = image_path.replace("JPEGImages", "depthImages") |