(self, batch)
| 69 | return self.outputs[0]['shape'], self.outputs[0]['dtype'] |
| 70 | |
| 71 | def infer(self, batch): |
| 72 | # Process I/O and execute the network |
| 73 | input = {self.inputs[0]['name']: tf.convert_to_tensor(batch)} |
| 74 | output = self.pred_fn(**input) |
| 75 | return output |
| 76 | |
| 77 | def process(self, batch, scales=None, nms_threshold=None): |
| 78 | # Infer network |