94 lines
2.9 KiB
JavaScript
94 lines
2.9 KiB
JavaScript
//
|
|
// Shinobi - Yolo Plugin
|
|
// Copyright (C) 2016-2025 Moe Alam, moeiscool
|
|
//
|
|
// # Donate
|
|
//
|
|
// If you like what I am doing here and want me to continue please consider donating :)
|
|
// PayPal : paypal@m03.ca
|
|
//
|
|
// Base Init >>
|
|
var fs = require('fs');
|
|
var config = require('./conf.json')
|
|
var s
|
|
try{
|
|
s = require('../pluginBase.js')(__dirname,config)
|
|
}catch(err){
|
|
console.log(err)
|
|
try{
|
|
s = require('./pluginBase.js')(__dirname,config)
|
|
}catch(err){
|
|
console.log(err)
|
|
return console.log(config.plug,'Plugin start has failed. pluginBase.js was not found.')
|
|
}
|
|
}
|
|
// Base Init />>
|
|
var yolo = require('node-yolo-shinobi');//this is @vapi/node-yolo@1.2.4 without the console output for detection speed
|
|
// var yolo = require('@vapi/node-yolo');
|
|
var detector = new yolo(__dirname + "/models", "cfg/coco.data", "cfg/yolov3.cfg", "yolov3.weights");
|
|
s.detectObject=function(buffer,d,tx,frameLocation){
|
|
var detectStuff = function(frame,callback){
|
|
detector.detect(frame)
|
|
.then(detections => {
|
|
matrices = []
|
|
detections.forEach(function(v){
|
|
matrices.push({
|
|
x:v.box.x,
|
|
y:v.box.y,
|
|
width:v.box.w,
|
|
height:v.box.h,
|
|
tag:v.className,
|
|
confidence:v.probability,
|
|
})
|
|
})
|
|
if(matrices.length > 0){
|
|
tx({
|
|
f:'trigger',
|
|
id:d.id,
|
|
ke:d.ke,
|
|
details:{
|
|
plug:config.plug,
|
|
name:'yolo',
|
|
reason:'object',
|
|
matrices:matrices,
|
|
imgHeight:parseFloat(d.mon.detector_scale_y),
|
|
imgWidth:parseFloat(d.mon.detector_scale_x)
|
|
}
|
|
})
|
|
}
|
|
fs.unlink(frame,function(){
|
|
|
|
})
|
|
})
|
|
.catch(error => {
|
|
console.log(error)
|
|
|
|
// here you can handle the errors. Ex: Out of memory
|
|
})
|
|
}
|
|
if(frameLocation){
|
|
detectStuff(frameLocation)
|
|
}else{
|
|
d.tmpFile=s.gid(5)+'.jpg'
|
|
if(!fs.existsSync(s.dir.streams)){
|
|
fs.mkdirSync(s.dir.streams);
|
|
}
|
|
d.dir=s.dir.streams+d.ke+'/'
|
|
if(!fs.existsSync(d.dir)){
|
|
fs.mkdirSync(d.dir);
|
|
}
|
|
d.dir=s.dir.streams+d.ke+'/'+d.id+'/'
|
|
if(!fs.existsSync(d.dir)){
|
|
fs.mkdirSync(d.dir);
|
|
}
|
|
fs.writeFile(d.dir+d.tmpFile,buffer,function(err){
|
|
if(err) return s.systemLog(err);
|
|
try{
|
|
detectStuff(d.dir+d.tmpFile)
|
|
}catch(error){
|
|
console.error('Catch: ' + error);
|
|
}
|
|
})
|
|
}
|
|
}
|