yolov5-eval

yolov5 的 npm 包，可以使用这个包结合yolov5的训练权重文件，部署到node环境

Usage

const yolo = require('yolov5-eval');
const path = require('path');

let source = path.resolve(__dirname, './inference/images'),
    weights = path.resolve(__dirname, './weights/v5-2/last.pt'),
    imgsz = 640;

let result = yolo.detectImage({image:source, weights, imgsz});

console.log(result);

const image = fs.readFileSync(source);
result = yolo.detectImage({image, weights, imgsz});

console.log(result);