Create loadMNISTLabels.m

llp1992 · llp1992 · commit 06cc99ed86b5 · 2016-05-02T15:11:28.000+08:00
diff --git a/DeepLearning/UFLDL/stl_exercise/loadMNISTLabels.m b/DeepLearning/UFLDL/stl_exercise/loadMNISTLabels.m
@@ -0,0 +1,19 @@
+function labels = loadMNISTLabels(filename)
+%loadMNISTLabels returns a [number of MNIST images]x1 matrix containing
+%the labels for the MNIST images
+
+fp = fopen(filename, 'rb');
+assert(fp ~= -1, ['Could not open ', filename, '']);
+
+magic = fread(fp, 1, 'int32', 0, 'ieee-be');
+assert(magic == 2049, ['Bad magic number in ', filename, '']);
+
+numLabels = fread(fp, 1, 'int32', 0, 'ieee-be');
+
+labels = fread(fp, inf, 'unsigned char');
+
+assert(size(labels,1) == numLabels, 'Mismatch in label count');
+
+fclose(fp);
+
+end
