1. 程式人生 > >MNIST 資料集視覺化程式碼

MNIST 資料集視覺化程式碼

寫一個matlab小程式將mnist資料集視覺化,將以下程式碼命名為image_visualization 並放在 $caffe_root/data/mnist/  下,{獨立執行,不必編譯caffe,但是要提前Linux下下載資料}

clear;
clc;
clear all;
image_file_name='t10k-images-idx3-ubyte';
index_filename='t10k-labels-idx1-ubyte';

fid1=fopen(image_file_name,'rb');
fid2=fopen(index_filename,'rb');

images_data=fread(fid1,'uint8');
index_data=fread(fid2,'uint8');

fclose(fid1);
fclose(fid2);

images_data=images_data(17:end);
index_data=index_data(9:end);
image_buffer=zeros(28,28);

for k=1:144:length(images_data)/28/28
    figure(144);
    for t=1:144
        image_buffer=reshape(images_data((k+t-2)*28*28+1:(k+t-1)*28*28),28,28);
        subplot(12,12,t);
        imshow(uint8(image_buffer)');
        title(num2str(index_data(k+t-1)));
    end
    pause;
end

結果如圖(100個數,原始碼是144個數,未貼出)