Downloads · 30 days
0
swin-ir-onnx
rocca/swin-ir-onnx
swin-ir-onnx is a machine learning model from rocca. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Here's what I did to export the pth to onnx (if only for my own future reference): 1. Open the Colab notebook and click Runtime Run All. 2. Open up and maintestswinir.py in the Colab editor and placing the following l…
Access
Public
Updated Apr 14, 2022
Repo size
61.3 MB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.onnx61.3 MB · 100%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Apr 14, 2022
- Updated
- Apr 14, 2022
- SHA
- 63b54415
Model card
From the Hugging Face model README
Here's what I did to export the pth to onnx (if only for my own future reference):
- Open the Colab notebook and click Runtime > Run All.
- Open up and
main_test_swinir.pyin the Colab editor and placing the following line afteroutput = model(img_lq):
torch.onnx.export(model, img_lq, "003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN-dynamic.onnx", export_params=True, opset_version=12, do_constant_folding=True, verbose=True, input_names = ['input'], output_names = ['output'], dynamic_axes={'input' : {2 : 'h', 3 : 'w'}, 'output' : {2 : 'h', 3 : 'w'}})
- Run this:
!python SwinIR/main_test_swinir.py --task real_sr --model_path experiments/pretrained_models/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq BSRGAN/testsets/RealSRSet --scale 4
And the ONNX file that you see in this repo will be generated.
- License
- apache-2.0
- Created
- Apr 14, 2022
- Updated
- Apr 14, 2022