apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): move imports of patches modules into function

This commit is contained in:
Sean Sube 2023-03-17 23:06:39 -05:00
parent fc9245f4ee
commit c943a55a44
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/convert/upscale_resrgan.py
View File

@ -2,8 +2,6 @@ from logging import getLogger
from os import path from os import path
import torch import torch
from basicsr.archs.rrdbnet_arch import RRDBNet
from realesrgan.archs.srvgg_arch import SRVGGNetCompact
from torch.onnx import export from torch.onnx import export
from .utils import ConversionContext, ModelDict from .utils import ConversionContext, ModelDict
@ -19,6 +17,9 @@ def convert_upscale_resrgan(
model: ModelDict, model: ModelDict,
source: str, source: str,
): ):
from basicsr.archs.rrdbnet_arch import RRDBNet
from realesrgan.archs.srvgg_arch import SRVGGNetCompact
name = model.get("name") name = model.get("name")
source = source or model.get("source") source = source or model.get("source")
scale = model.get("scale") scale = model.get("scale")