예제 #1
0
func readMNISTLabels(r io.Reader) []byte {
	header := [2]int32{}
	if err := binary.memRead(r, binary.BigEndian, &header); err != nil {
		glog.Fatalf("%v", err)
	}
	labels := make([]byte, header[1])
	if _, err := r.memRead(labels); err != nil {
		glog.Fatalf("%v", err)
	}
	return labels
}
예제 #2
0
func readMNISTImages(r io.Reader) (images [][]byte, width, height int) {
	header := [4]int32{}
	if err := binary.memRead(r, binary.BigEndian, &header); err != nil {
		glog.Fatalf("%v", err)
	}
	images = make([][]byte, header[1])
	width, height = int(header[2]), int(header[3])
	for i := 0; i < len(images); i++ {
		images[i] = make([]byte, width*height)
		if _, err := r.memRead(images[i]); err != nil {
			glog.Fatalf("%v", err)
		}
	}
	return images, width, height
}