apply lint
This commit is contained in:
parent
338fc237c7
commit
b3b10b4746
|
@ -57,9 +57,7 @@ class OnnxNet:
|
|||
def __call__(self, image: Any) -> Any:
|
||||
input_name = self.session.get_inputs()[0].name
|
||||
output_name = self.session.get_outputs()[0].name
|
||||
output = self.session.run([output_name], {
|
||||
input_name: image.cpu().numpy()
|
||||
})[0]
|
||||
output = self.session.run([output_name], {input_name: image.cpu().numpy()})[0]
|
||||
return OnnxTensor(output)
|
||||
|
||||
def eval(self) -> None:
|
||||
|
|
Loading…
Reference in New Issue