apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

switch to internal RRDBNet with x2 support

This commit is contained in:
Sean Sube 2023-12-29 08:50:27 -06:00
parent 1035915d36
commit da6beb31b2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/convert/upscaling/resrgan.py
View File

@ -5,6 +5,7 @@ from re import compile
import torch import torch
from torch.onnx import export from torch.onnx import export
from ...models.rrdb import RRDBNet
from ...models.srvgg import SRVGGNetCompact from ...models.srvgg import SRVGGNetCompact
from ..utils import ConversionContext, ModelDict from ..utils import ConversionContext, ModelDict
@ -61,8 +62,6 @@ def convert_upscale_resrgan(
model: ModelDict, model: ModelDict,
source: str, source: str,
): ):
from basicsr.archs.rrdbnet_arch import RRDBNet
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")