--- a
+++ b/partitionCEData.m
@@ -0,0 +1,33 @@
+% Developer: Tonmoy Ghosh (tghosh@crimson.ua.edu)
+function [imdsTrain, imdsTest, pxdsTrain, pxdsTest] = partitionCEData(imds,pxds)
+% Partition CamVid data by randomly selecting 60% of the data for training. The
+% rest is used for testing.
+
+% Set initial random state for example reproducibility.
+rng(0);
+numFiles = numel(imds.Files);
+shuffledIndices = randperm(numFiles);
+
+% Use 60% of the images for training.
+N = round(0.60 * numFiles);
+trainingIdx = shuffledIndices(1:N);
+
+% Use the rest for testing.
+testIdx = shuffledIndices(N+1:end);
+
+% Create image datastores for training and test.
+trainingImages = imds.Files(trainingIdx);
+testImages = imds.Files(testIdx);
+imdsTrain = imageDatastore(trainingImages);
+imdsTest = imageDatastore(testImages);
+
+% Extract class and label IDs info.
+classes = pxds.ClassNames;
+labelIDs = 1:numel(pxds.ClassNames);
+
+% Create pixel label datastores for training and test.
+trainingLabels = pxds.Files(trainingIdx);
+testLabels = pxds.Files(testIdx);
+pxdsTrain = pixelLabelDatastore(trainingLabels, classes, labelIDs);
+pxdsTest = pixelLabelDatastore(testLabels, classes, labelIDs);
+end