1 files changed, 52 insertions, 0 deletions
diff --git a/macros/partition.sci b/macros/partition.sci
new file mode 100644
index 0000000..b830597
--- /dev/null
+++ b/macros/partition.sci
@@ -0,0 +1,52 @@
+function [varargout]=partition(imageSet,groupSizesOrPercentages,varargin)
+// This function is used to split an imageSet.
+//
+// Calling Sequence
+// [set1 set2 ... setN] = partition(imgSet, groupSizes)
+// [set1 set2 ... setN] = partition(imgSet, groupPercentages)
+// [set1 set2 ... setN] = partition(..., method)
+//
+// Parameters
+// set: Output imageSet structure
+// imgSet: Input imageSet to be split
+// groupSizes: Matrix that specifies size of output sets
+// groupPercentages: Matrix that specifies percentage content of each output set
+// method: (Optional) Determines whether split is 'sequential' or 'randomized' (Default: 'sequential')
+//
+// Description
+// This function splits an imageSet into two or more imageSets on the basis of the groupSizes or groupPercentages provided.
+//
+// Examples
+// imgSet = imageSet(directory,'recursive');
+// [trainingSet testSet] = partition(imgSet,[0.8]);
+//
+// imgSet = imageSet(directory,'recursive');
+// [trainingSet testSet] = partition(imgSet,[5], 'randomized');
+//
+// Authors
+//  Rohit Suri
+
+    [lhs rhs]=argn(0);
+    if rhs>3
+        error(msprintf(" Too many input arguments"));
+    end
+    cols=size(groupSizesOrPercentages,'c');
+    if lhs>cols+1 then
+        error(msprintf(" Too many output arguments"));
+    end
+    imgSetList=imageSetToList(imageSet);
+    //sets=list();`
+    select rhs
+        case 2 then
+            sets=opencv_partition(imgSetList,groupSizesOrPercentages);
+        case 3 then
+            sets=opencv_partition(imgSetList,groupSizesOrPercentages,varargin(1));
+    end
+    for i=1:lhs
+        for j=1:size(sets(i)(4)(1),'r')
+                imgLocations(j)=sets(i)(4)(1)(j,1);
+        end
+        varargout(i)=struct('Description',sets(i)(2),'ImageLocation',imgLocations,'Count',int32(sets(i)(3)));
+	imgLocations=[]
+    end
+endfunction