diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-08-29 22:57:27 -0700 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-09-03 10:59:24 -0700 |
| commit | f2f73cfda455d941ace64294024042c602806639 (patch) | |
| tree | 1bc8f41d4a3b67ea1d9230d67ce8c0e233f5eed9 /include | |
| parent | c3151bb9aed7deb8b7483f46e7e36bff80efca87 (diff) | |
| download | caffeonacl-f2f73cfda455d941ace64294024042c602806639.tar.gz caffeonacl-f2f73cfda455d941ace64294024042c602806639.tar.bz2 caffeonacl-f2f73cfda455d941ace64294024042c602806639.zip | |
vision_layers.hpp: Doxygen \brief & TODO stubs.
Diffstat (limited to 'include')
| -rw-r--r-- | include/caffe/vision_layers.hpp | 32 |
1 files changed, 23 insertions, 9 deletions
diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp index 02f9e43b..86a54ced 100644 --- a/include/caffe/vision_layers.hpp +++ b/include/caffe/vision_layers.hpp @@ -16,8 +16,12 @@ namespace caffe { -/* ConvolutionLayer -*/ +/** + * @brief Convolves the input image with a bank of learned filters, + * and (optionally) adds biases. + * + * TODO(dox): thorough documentation for Forward, Backward, and proto params. + */ template <typename Dtype> class ConvolutionLayer : public Layer<Dtype> { public: @@ -60,8 +64,13 @@ class ConvolutionLayer : public Layer<Dtype> { int N_; }; -/* Im2colLayer -*/ +/** + * @brief A helper for image operations that rearranges image regions into + * column vectors. Used by ConvolutionLayer to perform convolution + * by matrix multiplication. + * + * TODO(dox): thorough documentation for Forward, Backward, and proto params. + */ template <typename Dtype> class Im2colLayer : public Layer<Dtype> { public: @@ -98,9 +107,11 @@ class Im2colLayer : public Layer<Dtype> { template <typename Dtype> class PoolingLayer; template <typename Dtype> class SplitLayer; -/* LRNLayer - Local Response Normalization -*/ +/** + * @brief Normalize the input in a local region across or within feature maps. + * + * TODO(dox): thorough documentation for Forward, Backward, and proto params. + */ template <typename Dtype> class LRNLayer : public Layer<Dtype> { public: @@ -171,8 +182,11 @@ class LRNLayer : public Layer<Dtype> { }; -/* PoolingLayer -*/ +/** + * @brief Pools the input image by taking the max, average, etc. within regions. + * + * TODO(dox): thorough documentation for Forward, Backward, and proto params. + */ template <typename Dtype> class PoolingLayer : public Layer<Dtype> { public: |