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);