darknet/spec/darknet.spec.js

A Node wrapper of pjreddie's open source neural network framework Darknet, using the Foreign Function Interface Library. Read: YOLOv3 in JavaScript.

const { Darknet } = require('../darknet');
const path = require('path');

const config = {
    weights: path.resolve(__dirname + '/../examples/yolov3-tiny.weights'),
    config: path.resolve(__dirname + '/../examples/yolov3-tiny.cfg'),
    namefile: path.resolve(__dirname + '/../examples/coco.names')
};

const DOG_RESULT = 
[ 
    { 
        name: 'car',
        prob: 0.6152912378311157,
        box: 
        { 
            x: 572.1994018554688,
            y: 120.48184204101562,
            w: 214.3546600341797,
            h: 98.72494506835938 
        } 
    },
    { 
        name: 'bicycle',
        prob: 0.5850223302841187,
        box: 
        {
            x: 390.94427490234375,
            y: 300.541259765625,
            w: 369.40216064453125,
            h: 299.3248291015625 
        } 
    },
    { name: 'dog',
        prob: 0.5707316994667053,
        box: 
        { 
            x: 249.094970703125,
            y: 352.12335205078125,
            w: 239.8490753173828,
            h: 330.763671875 
        } 
    }
];

const EAGLE_RESULT =
[ 
    { 
        name: 'bird',
        prob: 0.7649939656257629,
        box: 
        { 
            x: 379.6971130371094,
            y: 260.68951416015625,
            w: 486.4349365234375,
            h: 322.6314392089844 
        } 
    } 
];

const GIRAFFE_RESULT =
[ 
    { 
        name: 'zebra',
        prob: 0.5318787097930908,
        box: 
        { 
            x: 359.7569885253906,
            y: 326.3610534667969,
            w: 109.50553894042969,
            h: 242.93246459960938 
        } 
    },
    { 
        name: 'giraffe',
        prob: 0.528597891330719,
        box: 
        { 
            x: 296.3047180175781,
            y: 210.2434844970703,
            w: 285.6275329589844,
            h: 504.72705078125 
        } 
    } 
];

describe('darknet', () => {

    let darknet;

    it('detects various images', () => {
        darknet = new Darknet(config);

        const dog = darknet.detect(image('dog.jpg'));
        expect(dog).toEqual(DOG_RESULT);

        const eagle = darknet.detect(image('eagle.jpg'));
        expect(eagle).toEqual(EAGLE_RESULT);

        const giraffe = darknet.detect(image('giraffe.jpg'));
        expect(giraffe).toEqual(GIRAFFE_RESULT);
    });

    // xit('detects various images async (concurrent)', async () => {
    //     darknet_a = new Darknet(config);
    //     darknet_b = new Darknet(config);
    //     darknet_c = new Darknet(config);

    //     return Promise.all([
    //         darknet_a.detectAsync(image('dog.jpg')),
    //         darknet_b.detectAsync(image('eagle.jpg')),
    //         darknet_c.detectAsync(image('giraffe.jpg'))
    //     ]).then(values => {
    //             expect(JSON.stringify(values[0])).toBe(JSON.stringify(DOG_RESULT));
    //             expect(JSON.stringify(values[1])).toBe(JSON.stringify(EAGLE_RESULT));
    //             expect(JSON.stringify(values[2])).toBe(JSON.stringify(GIRAFFE_RESULT));
    //         });
    // });

    // describe('experimental', () => {
    //     xit('detects images async', async () => {
    //         darknet = new DarknetExperimental(config);
    //         return Promise.all([
    //             darknet.detectAsync(image('dog.jpg')),
    //             darknet.detectAsync(image('eagle.jpg')),
    //             darknet.detectAsync(image('giraffe.jpg'))
    //         ]).then(values => {
    //             expect(JSON.stringify(values[0])).toBe(JSON.stringify(DOG_RESULT));
    //             expect(JSON.stringify(values[1])).toBe(JSON.stringify(EAGLE_RESULT));
    //             expect(JSON.stringify(values[2])).toBe(JSON.stringify(GIRAFFE_RESULT));
    //         });
    //     });

    // })

});

function image(location) {
    return path.resolve(__dirname + '/../examples/' + location);
}