aboutsummaryrefslogtreecommitdiff
path: root/examples/graph_edsr.cpp
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2020-08-26 14:48:57 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2020-09-08 15:54:26 +0000
commite4340a4afe6c5ca35fb2ce280152c6504a88cf21 (patch)
tree6ac5bb184890faaabbecc51e3a25a068ff5c5bf3 /examples/graph_edsr.cpp
parent4e28d60943551e400ee202087ab1c8bc022dad9f (diff)
downloadComputeLibrary-e4340a4afe6c5ca35fb2ce280152c6504a88cf21.tar.gz
COMPMID-3568: Add provenance of model for EDSR graph example
Change-Id: I0c2404b228eaf7e6fca9d393da4df5610fda4482 Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3936 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'examples/graph_edsr.cpp')
-rw-r--r--examples/graph_edsr.cpp21
1 files changed, 21 insertions, 0 deletions
diff --git a/examples/graph_edsr.cpp b/examples/graph_edsr.cpp
index 3868f0f7c4..77783d97ed 100644
--- a/examples/graph_edsr.cpp
+++ b/examples/graph_edsr.cpp
@@ -102,6 +102,27 @@ private:
GraphEdsr model{};
};
+/** Internal implementation of UINT8 EDSR with some modifications from the paper.
+ * The sub-pixel convolution has been replaced with a deconvolution layer. This
+ * operation is mathematically the same.
+ *
+ * Convolution replaced by deconvolution:
+ * https://arxiv.org/abs/1609.07009
+ * "Is the deconvolution layer the same as a convolutional layer?"
+ * Wenzhe Shi, Jose Caballero, Lucas Theis, Ferenc Huszar, Andrew Aitken, Christian Ledig, Zehan Wang
+ *
+ * Original model is:
+ * https://arxiv.org/abs/1707.02921
+ * "Enhanced Deep Residual Networks for Single Image Super-Resolution"
+ * Bee Lim, Sanghyun Son, Heewon Kim, Seungjun Nah, Kyoung Mu Lee
+ *
+ * @note To list all the possible arguments execute the binary appended with the --help option
+ *
+ * @param[in] argc Number of arguments
+ * @param[in] argv Arguments
+ *
+ * @return Return code
+ */
int main(int argc, char **argv)
{
return run_example<GraphEdsrExample>(argc, argv);