(self, args)
| 73 | |
| 74 | class Predictor(object): |
| 75 | def __init__(self, args): |
| 76 | self.rgb_means = (0.485, 0.456, 0.406) |
| 77 | self.std = (0.229, 0.224, 0.225) |
| 78 | self.args = args |
| 79 | self.session = onnxruntime.InferenceSession(args.model) |
| 80 | self.input_shape = tuple(map(int, args.input_shape.split(','))) |
| 81 | |
| 82 | def inference(self, ori_img, timer): |
| 83 | img_info = {"id": 0} |
nothing calls this directly
no outgoing calls
no test coverage detected