Skip to content

Commit 4201327

Browse files
author
Lara
committed
add info about the model
1 parent 28528e0 commit 4201327

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

advanced_source/super_resolution_with_onnxruntime.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,17 @@
3535
######################################################################
3636
# Super-resolution is a way of increasing the resolution of images, videos
3737
# and is widely used in image processing or video editing. For this
38-
# tutorial, we will first use a small super-resolution model with a dummy
39-
# input.
38+
# tutorial, we will use a small super-resolution model.
4039
#
41-
# First, let's create a SuperResolution model in PyTorch. `This
40+
# First, let's create a SuperResolution model in PyTorch.
41+
# This model uses the efficient sub-pixel convolution layer described in
42+
# `"Real-Time Single Image and Video Super-Resolution Using an Efficient
43+
# Sub-Pixel Convolutional Neural Network" - Shi et al <https://arxiv.org/abs/1609.05158>`__
44+
# for increasing the resolution of an image by an upscale factor.
45+
# The model expects the Y component of the YCbCr of an image as an input, and
46+
# outputs the upscaled Y component in super resolution.
47+
#
48+
# `The
4249
# model <https://github.com/pytorch/examples/blob/master/super_resolution/model.py>`__
4350
# comes directly from PyTorch's examples without modification:
4451
#

0 commit comments

Comments
 (0)