Augmentation work

augmentation/augmentation.hpp

@@ -42,10 +42,6 @@ class Augmentation
    *                                the dataset.
    *                                NOTE : This doesn't apply to augmentations
    *                                such as resize.
-   * @param batches Boolean to determine if input is a single data point or
-   *                a batch. Defaults to true.
-   *                NOTE : If true, each data point must be represented as a
-   *                seperate column.
    */
   Augmentation(const std::vector<std::string>& augmentation,
                const double augmentationProbability);
@@ -127,4 +123,6 @@ class Augmentation
   double augmentationProbability;
 };
 
+#include "augmentation_impl.hpp" // Include implementation.
+
 #endif

augmentation/augmentation_impl.hpp

@@ -0,0 +1,24 @@
+/**
+ * @file augmentation_impl.hpp
+ * @author Kartik Dutt
+ * 
+ * Implementation of Augmentation class for augmenting data.
+ *
+ * mlpack is free software; you may redistribute it and/or modify it under the
+ * terms of the 3-clause BSD license.  You should have received a copy of the
+ * 3-clause BSD license along with mlpack.  If not, see
+ * http://www.opensource.org/licenses/BSD-3-Clause for more information.
+ */
+// Incase it has not been included already.
+#include "augmentation.hpp"
+
+#ifndef MODELS_AUGMENTATION_IMPL_HPP
+#define MODELS_AUGMENTATION_IMPL_HPP
+
+Augmentation::Augmentation() :
+    augmentations(std::vector<std::string>()),
+    augmentationProbability(0.2)
+{
+  // Nothing to do here.
+}
+#endif

dataloader/dataloader_impl.hpp

@@ -216,6 +216,8 @@ template<
     size_t imageDepth = std::stoi(sizeInformation.get_child("depth").data());
     mlpack::data::ImageInfo imageInfo(imageWidth, imageHeight, imageDepth);
 
+    // TODO: Resize the image here.
+
     // Load the image.
     // The image loaded here will be in column format i.e. Output will be matrix with the
     // following shape {1, cols * rows * slices} in column major format.
@@ -250,6 +252,8 @@ template<
         }
       }
     }
+
+    // TODO: Augment the image here.
   }
 }