Skip to content

Instantly share code, notes, and snippets.

@bakoushin
Last active June 18, 2020 17:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bakoushin/a1b16f024e5b5bec7af0b9f56b904b47 to your computer and use it in GitHub Desktop.
Save bakoushin/a1b16f024e5b5bec7af0b9f56b904b47 to your computer and use it in GitHub Desktop.
async function* dataGenerator() {
while (true) {
const csvStream = fs.createReadStream(pathToCSV).pipe(csv({
headers: ['center', 'left', 'right', 'steering', 'throttle', 'brake', 'speed'],
mapValues: ({ value }) => value.trim()
}));
for await (const { center, left, right, steering } of csvStream) {
const centerImageBuffer = fs.promises.readFile(center);
const leftImageBuffer = fs.promises.readFile(left);
const rightImageBuffer = fs.promises.readFile(right);
const offset = 0.333;
yield [await centerImageBuffer, Number(steering)];
yield [await leftImageBuffer, Number(steering) + offset];
yield [await rightImageBuffer, Number(steering) - offset];
}
csvStream.destroy();
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment