apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-02-18 08:46:46 -06:00
parent 338fc237c7
commit b3b10b4746
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/onnx/onnx_net.py
View File

@ -57,9 +57,7 @@ class OnnxNet:
def __call__(self, image: Any) -> Any: def __call__(self, image: Any) -> Any:
input_name = self.session.get_inputs()[0].name input_name = self.session.get_inputs()[0].name
output_name = self.session.get_outputs()[0].name output_name = self.session.get_outputs()[0].name
output = self.session.run([output_name], { output = self.session.run([output_name], {input_name: image.cpu().numpy()})[0]
input_name: image.cpu().numpy()
})[0]
return OnnxTensor(output) return OnnxTensor(output)
def eval(self) -> None: def eval(self) -> None: