MCPcopy Create free account
hub / github.com/tensorflow/tfjs-examples / loadImages

Function loadImages

mnist-node/data.js:72–97  ·  view source on GitHub ↗
(filename)

Source from the content-addressed store, hash-verified

70}
71
72async function loadImages(filename) {
73 const buffer = await fetchOnceAndSaveToDiskWithBuffer(filename);
74
75 const headerBytes = IMAGE_HEADER_BYTES;
76 const recordBytes = IMAGE_HEIGHT * IMAGE_WIDTH;
77
78 const headerValues = loadHeaderValues(buffer, headerBytes);
79 assert.equal(headerValues[0], IMAGE_HEADER_MAGIC_NUM);
80 assert.equal(headerValues[2], IMAGE_HEIGHT);
81 assert.equal(headerValues[3], IMAGE_WIDTH);
82
83 const images = [];
84 let index = headerBytes;
85 while (index < buffer.byteLength) {
86 const array = new Float32Array(recordBytes);
87 for (let i = 0; i < recordBytes; i++) {
88 // Normalize the pixel values into the 0-1 interval, from
89 // the original 0-255 interval.
90 array[i] = buffer.readUInt8(index++) / 255;
91 }
92 images.push(array);
93 }
94
95 assert.equal(images.length, headerValues[1]);
96 return images;
97}
98
99async function loadLabels(filename) {
100 const buffer = await fetchOnceAndSaveToDiskWithBuffer(filename);

Callers 1

loadDataMethod · 0.70

Calls 2

loadHeaderValuesFunction · 0.70

Tested by

no test coverage detected