Helper class to store detection results from ONNX model.
| 52 | |
| 53 | |
| 54 | class YoloResult: |
| 55 | """Helper class to store detection results from ONNX model.""" |
| 56 | |
| 57 | def __init__(self, boxes, names): |
| 58 | self.boxes = [YoloBox(data=d) for d in boxes] |
| 59 | self.boxes.sort(key=lambda x: x.conf, reverse=True) |
| 60 | self.names = names |
| 61 | |
| 62 | |
| 63 | class YoloBox: |
no outgoing calls