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

2
api/onnx_web/chain/upscale_resrgan.py
View File

@ -46,7 +46,7 @@ def load_resrgan(
sess_options=device.sess_options(), sess_options=device.sess_options(),
) )
elif params.format == "pth": elif params.format == "pth":
if TAG_X4_V3 in model_file: if TAG_X4_V3 in model_file:
# the x4-v3 model needs a different network # the x4-v3 model needs a different network
model = SRVGGNetCompact( model = SRVGGNetCompact(
num_in_ch=3, num_in_ch=3,

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: