(self)
| 69 | self.set_transform() |
| 70 | |
| 71 | def set_transform(self): |
| 72 | self.transform = tf.Compose([ |
| 73 | tf.Resize(self.opt.down_resolution), |
| 74 | tf.ToTensor(), |
| 75 | ]) |
| 76 | self.depth_transform = tf.Compose([ |
| 77 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 78 | tf.ToTensor(), |
| 79 | ]) |
| 80 | self.mask_transform = tf.Compose([ |
| 81 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 82 | ]) |
| 83 | |
| 84 | def transform_mask(self, mask): |
| 85 | mask = self.mask_transform(mask) |