| 81 | self.ret_frames.append((chunk_idx, i_vid, i_frame)) |
| 82 | |
| 83 | def set_transform(self): |
| 84 | self.transform = tf.Compose([ |
| 85 | tf.Resize(self.opt.down_resolution), |
| 86 | tf.ToTensor(), |
| 87 | ]) |
| 88 | self.depth_transform = tf.Compose([ |
| 89 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 90 | tf.ToTensor(), |
| 91 | ]) |
| 92 | self.depth_transform_gt = tf.Compose([ |
| 93 | tf.Resize(self.opt.down_resolution, interpolation=tf.InterpolationMode.NEAREST), |
| 94 | tf.ToTensor(), |
| 95 | ]) |
| 96 | |
| 97 | self.to_tensor = tf.ToTensor() |
| 98 | self.down_resize = tf.Resize(self.opt.down_resolution) |
| 99 | |
| 100 | def __len__(self): |
| 101 | return len(self.ret_frames) |