java解析bmp文件格式_在Java中读取IDX文件类型

非常直截了当,正如WPrecht所说：“URL描述了你必须解码的格式”.这是我的idx文件的ImageSet导出器,不是很干净,但它做了它必须做的事情.

public class IdxReader {

public static void main(String[] args) {

// TODO Auto-generated method stub

FileInputStream inImage = null;

FileInputStream inLabel = null;

String inputImagePath = "CBIR_Project/imagesRaw/MNIST/train-images-idx3-ubyte";

String inputLabelPath = "CBIR_Project/imagesRaw/MNIST/train-labels-idx1-ubyte";

String outputPath = "CBIR_Project/images/MNIST_Database_ARGB/";

int[] hashMap = new int[10];

try {

inImage = new FileInputStream(inputImagePath);

inLabel = new FileInputStream(inputLabelPath);

int magicNumberImages = (inImage.read() << 24) | (inImage.read() << 16) | (inImage.read() << 8) | (inImage.read());

int numberOfImages = (inImage.read() << 24) | (inImage.read() << 16) | (inImage.read() << 8) | (inImage.read());

int numberOfRows = (inImage.read() << 24) | (inImage.read() << 16) | (inImage.read() << 8) | (inImage.read());

int numberOfColumns = (inImage.read() << 24) | (inImage.read() << 16) | (inImage.read() << 8) | (inImage.read());

int magicNumberLabels = (inLabel.read() << 24) | (inLabel.read() << 16) | (inLabel.read() << 8) | (inLabel.read());

int numberOfLabels = (inLabel.read() << 24) | (inLabel.read() << 16) | (inLabel.read() << 8) | (inLabel.read());

BufferedImage image = new BufferedImage(numberOfColumns, numberOfRows, BufferedImage.TYPE_INT_ARGB);

int numberOfPixels = numberOfRows * numberOfColumns;

int[] imgPixels = new int[numberOfPixels];

for(int i = 0; i < numberOfImages; i++) {

if(i % 100 == 0) {System.out.println("Number of images extracted: " + i);}

for(int p = 0; p < numberOfPixels; p++) {

int gray = 255 - inImage.read();

imgPixels[p] = 0xFF000000 | (gray<<16) | (gray<<8) | gray;

}

image.setRGB(0, 0, numberOfColumns, numberOfRows, imgPixels, 0, numberOfColumns);

int label = inLabel.read();

hashMap[label]++;

File outputfile = new File(outputPath + label + "_0" + hashMap[label] + ".png");

ImageIO.write(image, "png", outputfile);

}

} catch (FileNotFoundException e) {

// TODO Auto-generated catch block

e.printStackTrace();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

} finally {

if (inImage != null) {

try {

inImage.close();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

}

if (inLabel != null) {

try {

inLabel.close();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

}

}

}

}