(self)
| 135 | return data |
| 136 | |
| 137 | def set_transform(self): |
| 138 | self.transform = tf.Compose([ |
| 139 | tf.Resize(self.opt.down_resolution), |
| 140 | tf.ToTensor(), |
| 141 | ]) |
| 142 | self.depth_transform = tf.Compose([ |
| 143 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 144 | tf.ToTensor(), |
| 145 | ]) |
| 146 | self.mask_transform = tf.Compose([ |
| 147 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 148 | ]) |