NodeJS script to sort CIFAR-10 dataset of PNGs into label directories

sort-cifar.js

var fs = require('fs')

var labels = fs.readFileSync('labels.txt', 'utf-8')
    .split('\n')
    .filter(Boolean);

/*for(i = 0; i < labels.length; i++) {
  fs.mkdirSync("./train/"+i);
  fs.mkdirSync("./test/"+i);
}*/

var set = './train/'

fs.readdirSync(set).forEach(file => {
  console.log(file);
  if(file.indexOf('.png') > -1) {
    var label = file.match(/([a-zA-Z0-9_-])\_(.*)\.png/).pop()
    var id = labels.indexOf(label)
    console.log('Label is: '+id)
    fs.rename(set+file, set+id+'/'+file)
    console.log('Moved to: '+set+id+'/'+file)
  }
})