Args: inputs: [2, C, H, W]
(self, inputs)
| 150 | tf.TensorSpec((1, 2, self.height, self.width), tf.float32, 'gt_flow')] |
| 151 | |
| 152 | def graph_structure(self, inputs): |
| 153 | """ |
| 154 | Args: |
| 155 | inputs: [2, C, H, W] |
| 156 | """ |
| 157 | raise NotImplementedError() |
| 158 | |
| 159 | def preprocess(self, left, right): |
| 160 | x = tf.concat([left, right], axis=0) # 2CHW |