node-yolo-shinobi/index.js

Node.js interface for Yolo/Darknet

const YoloModule = require(__dirname + '/build/Release/nodeyolojs').Yolo;
const imagemagick = require('imagickal');

class Yolo
{
 constructor(working_dir, dataset, configuration_file, weights)
 {
  this.yolo = new YoloModule(working_dir, dataset, configuration_file, weights);
 }

 detect(image_path)
 {
  return new Promise((resolve, reject) =>
  {
   imagemagick.identify(image_path, true).then(() =>
   {
    this.yolo.detect(image_path).then(detections =>
    {
     resolve(detections);
    })
    .catch(error =>
    {
     reject(error);
    });
   })
   .catch(() =>
   {
    reject("Image " + image_path + " is corrupted.");
   });
  });
 }
}

module.exports = Yolo;