(self, left, right, gt_flow)
| 165 | return resize(prediction * DISP_SCALE, mode='bilinear') |
| 166 | |
| 167 | def build_graph(self, left, right, gt_flow): |
| 168 | x = self.preprocess(left, right) |
| 169 | prediction = self.graph_structure(x) |
| 170 | prediction = self.postprocess(prediction) |
| 171 | tf.identity(prediction, name="prediction") |
| 172 | # endpoint error |
| 173 | tf.reduce_mean(tf.norm(prediction - gt_flow, axis=1), name='epe') |
| 174 | |
| 175 | |
| 176 | class FlowNet2(FlowNetBase): |
nothing calls this directly
no test coverage detected