(self, file, num_classes, target_dtype = np.float32)
| 26 | |
| 27 | class ONNXClassifierWrapper(): |
| 28 | def __init__(self, file, num_classes, target_dtype = np.float32): |
| 29 | |
| 30 | self.target_dtype = target_dtype |
| 31 | self.num_classes = num_classes |
| 32 | self.load(file) |
| 33 | |
| 34 | self.stream = None |
| 35 | |
| 36 | def load(self, file): |
| 37 | f = open(file, "rb") |