diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-08-26 14:48:57 +0100 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-09-08 15:54:26 +0000 |
commit | e4340a4afe6c5ca35fb2ce280152c6504a88cf21 (patch) | |
tree | 6ac5bb184890faaabbecc51e3a25a068ff5c5bf3 | |
parent | 4e28d60943551e400ee202087ab1c8bc022dad9f (diff) | |
download | ComputeLibrary-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>
-rw-r--r-- | examples/graph_edsr.cpp | 21 |
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); |