diff options
author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-08-29 22:48:11 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-09-03 10:59:24 -0700 |
commit | c3151bb9aed7deb8b7483f46e7e36bff80efca87 (patch) | |
tree | db43d3b54a7e1f3ce7531a6284480845fa1757ec /include | |
parent | 57171b507aff14f16773b4791b53d19aee1e7be8 (diff) | |
download | caffeonacl-c3151bb9aed7deb8b7483f46e7e36bff80efca87.tar.gz caffeonacl-c3151bb9aed7deb8b7483f46e7e36bff80efca87.tar.bz2 caffeonacl-c3151bb9aed7deb8b7483f46e7e36bff80efca87.zip |
data_layers: Doxygen \brief & TODO stubs.
Diffstat (limited to 'include')
-rw-r--r-- | include/caffe/data_layers.hpp | 43 |
1 files changed, 39 insertions, 4 deletions
diff --git a/include/caffe/data_layers.hpp b/include/caffe/data_layers.hpp index f972965b..bebbab10 100644 --- a/include/caffe/data_layers.hpp +++ b/include/caffe/data_layers.hpp @@ -23,8 +23,14 @@ namespace caffe { #define HDF5_DATA_DATASET_NAME "data" #define HDF5_DATA_LABEL_NAME "label" -// TODO: DataLayer, ImageDataLayer, and WindowDataLayer all have the -// same basic structure and a lot of duplicated code. +/** + * @brief Provides data to the Net from a LevelDB or LMDB. + * + * TODO: DataLayer, ImageDataLayer, and WindowDataLayer all have the + * same basic structure and a lot of duplicated code. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class DataLayer : public Layer<Dtype>, public InternalThread { public: @@ -80,6 +86,11 @@ class DataLayer : public Layer<Dtype>, public InternalThread { Caffe::Phase phase_; }; +/** + * @brief Provides data to the Net generated by a Filler. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class DummyDataLayer : public Layer<Dtype> { public: @@ -106,6 +117,11 @@ class DummyDataLayer : public Layer<Dtype> { vector<bool> refill_; }; +/** + * @brief Provides data to the Net from HDF5 files. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class HDF5DataLayer : public Layer<Dtype> { public: @@ -140,6 +156,11 @@ class HDF5DataLayer : public Layer<Dtype> { Blob<Dtype> label_blob_; }; +/** + * @brief Write blobs to disk as HDF5 files. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class HDF5OutputLayer : public Layer<Dtype> { public: @@ -174,6 +195,11 @@ class HDF5OutputLayer : public Layer<Dtype> { Blob<Dtype> label_blob_; }; +/** + * @brief Provides data to the Net from image files. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class ImageDataLayer : public Layer<Dtype>, public InternalThread { public: @@ -221,8 +247,11 @@ class ImageDataLayer : public Layer<Dtype>, public InternalThread { Caffe::Phase phase_; }; -/* MemoryDataLayer -*/ +/** + * @brief Provides data to the Net from memory. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class MemoryDataLayer : public Layer<Dtype> { public: @@ -264,6 +293,12 @@ class MemoryDataLayer : public Layer<Dtype> { int pos_; }; +/** + * @brief Provides data to the Net from windows of images files, specified + * by a window data file. + * + * TODO(dox): thorough documentation for Forward and proto params. + */ template <typename Dtype> class WindowDataLayer : public Layer<Dtype>, public InternalThread { public: |