add PlateRecognizer plugin

plugins/platerecognizer/.gitignore

@@ -0,0 +1,4 @@
+conf.json
+dist
+models
+node_modules

plugins/platerecognizer/INSTALL.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+DIR=$(dirname $0)
+echo "Removing existing Node.js modules..."
+rm -rf $DIR/node_modules
+
+nonInteractiveFlag=false
+
+if [ ! -e "$DIR/conf.json" ]; then
+	dontCreateKeyFlag=false
+    echo "Creating conf.json"
+    sudo cp $DIR/conf.sample.json $DIR/conf.json
+else
+    echo "conf.json already exists..."
+fi
+
+if [ "$dontCreateKeyFlag" = false ]; then
+	echo "Adding Random Plugin Key to Main Configuration"
+	node $DIR/../../tools/modifyConfigurationForPlugin.js platerecognizer key=$(head -c 64 < /dev/urandom | sha256sum | awk '{print substr($1,1,60)}')
+fi

plugins/platerecognizer/conf.sample.json

@@ -0,0 +1,10 @@
+{
+    "plug": "PlateRecognizer",
+    "host": "localhost",
+    "platerecognizerApiKey": "11111111111111111",
+    "port": 8080,
+    "hostPort": 58084,
+    "key": "1234567890",
+    "mode": "client",
+    "type": "detector"
+}

plugins/platerecognizer/package.json

@@ -0,0 +1,26 @@
+{
+   "name": "shinobi-platerecognizer",
+   "author": "Elad Bar",
+   "version": "1.0.0",
+   "description": "Object Detection plugin for DeepStack",
+   "main": "shinobi-platerecognizer.js",
+   "dependencies": {
+      "request": "^2.88.0",
+      "express": "^4.16.2",
+      "moment": "^2.19.2",
+      "socket.io": "^2.3.0",
+      "socket.io-client": "^2.3.0"
+   },
+   "devDependencies": {},
+   "bin": "shinobi-platerecognizer.js",
+   "pkg": {
+      "targets": [
+         "node12"
+      ],
+      "scripts": [
+         "../pluginBase.js"
+      ],
+      "assets": []
+   },
+   "disabled": true
+}

plugins/platerecognizer/shinobi-platerecognizer.js

@@ -0,0 +1,192 @@
+//
+// Shinobi - Tensorflow Plugin
+// Copyright (C) 2016-2025 Elad Bar, Moe Alam
+//
+// Base Init >>
+const fs = require('fs');
+const config = require('./conf.json')
+const fetch = require('node-fetch');
+const FormData = require('form-data');
+var s
+const {
+		workerData
+	} = require('worker_threads');
+
+if(workerData && workerData.ok === true){
+	try{
+		s = require('../pluginWorkerBase.js')(__dirname,config)
+	}catch(err){
+		console.log(err)
+		try{
+			s = require('./pluginWorkerBase.js')(__dirname,config)
+		}catch(err){
+			console.log(err)
+			return console.log(config.plug,'WORKER : Plugin start has failed. pluginBase.js was not found.')
+		}
+	}
+}else{
+	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.')
+		}
+	}
+	try{
+		s = require('../pluginBase.js')(__dirname,config)
+	}catch(err){
+		console.log(err)
+		try{
+			const {
+				haltMessage,
+				checkStartTime,
+				setStartTime,
+			} = require('../pluginCheck.js')
+
+			if(!checkStartTime()){
+				console.log(haltMessage,new Date())
+				s.disconnectWebSocket()
+				return
+			}
+			setStartTime()
+		}catch(err){
+			console.log(`pluginCheck failed`)
+		}
+	}
+
+}
+// Base Init />>
+
+const licensePlateRegion = config.licensePlateRegion || 'us'
+const platerecognizerApiKey = config.platerecognizerApiKey || '111111111111111111'
+if(!config.platerecognizerApiKey){
+	console.log('No Plate Recognizer API Key set.')
+	console.log('set conf.json value for `platerecognizerApiKey`')
+	return process.exit()
+}
+const baseUrl = config.platerecognizerEndpoint || "https://api.platerecognizer.com/v1/plate-reader/"
+
+function platerecognizerRequest(d,frameBuffer){
+	return new Promise((resolve,reject) => {
+		try{
+			let body = new FormData();
+			frameBufferToPath(d,frameBuffer).then((filePath) => {
+				body.append('upload', fs.createReadStream(filePath));
+				// Or body.append('upload', base64Image);
+				body.append('regions', licensePlateRegion); // Change to your country
+				fetch(baseUrl, {
+			        method: 'POST',
+			        headers: {
+						"Authorization": `Token ${platerecognizerApiKey}`
+			        },
+			        body: body
+			    }).then(res => res.json())
+			    .then((json) => {
+					let predictions = []
+					try{
+						const response = json || {results: []}
+						predictions = response["results"] || []
+					}catch(err){
+						console.log(json)
+						console.log(err)
+						console.log(body)
+					}
+					resolve(predictions);
+					fs.unlink(filePath,function(){
+
+					})
+				})
+			    .catch((err) => {
+			        console.log(err);
+			    });
+			})
+		}catch(err){
+			resolve([])
+			console.log(err)
+		}
+	})
+}
+function addVehicleMatrix(v,mats){
+	const label = v.vehicle["type"]
+	const confidence = v.vehicle["score"]
+	const y_min = v.vehicle["ymin"]
+	const x_min = v.vehicle["xmin"]
+	const y_max = v.vehicle["ymax"]
+	const x_max = v.vehicle["xmax"]
+	const vehicleWidth = x_max - x_min
+	const vehicleHeight = y_max - y_min
+	mats.push({
+		x: x_min,
+		y: y_min,
+		width: vehicleWidth,
+		height: vehicleHeight,
+		tag: label,
+		confidence: confidence,
+	})
+}
+function frameBufferToPath(d,buffer){
+	return new Promise((resolve,reject) => {
+		const tmpFile = s.gid(5)+'.jpg'
+		if(!fs.existsSync(s.dir.streams)){
+			fs.mkdirSync(s.dir.streams);
+		}
+		frameDirectory = s.dir.streams+d.ke+'/'+d.id+'/'
+		fs.writeFile(frameDirectory+tmpFile,buffer,function(err){
+			if(err) return s.systemLog(err);
+			try{
+				resolve(frameDirectory+tmpFile)
+			}catch(error){
+				console.error('Catch: ' + error);
+			}
+		})
+	})
+}
+s.detectObject = async function(frameBuffer,d,tx,frameLocation,callback){
+	const timeStart = new Date()
+	const predictions = await platerecognizerRequest(d,frameBuffer)
+	if(predictions.length > 0) {
+		const mats = []
+		predictions.forEach(function(v){
+			const label = v["plate"]
+			const confidence = v["score"]
+			const y_min = v.box["ymin"]
+			const x_min = v.box["xmin"]
+			const y_max = v.box["ymax"]
+			const x_max = v.box["xmax"]
+			const width = x_max - x_min
+			const height = y_max - y_min
+			mats.push({
+				x: x_min,
+				y: y_min,
+				width: width,
+				height: height,
+				tag: label,
+				confidence: confidence,
+			})
+			addVehicleMatrix(v,mats)
+		})
+		const isObjectDetectionSeparate = d.mon.detector_pam === '1' && d.mon.detector_use_detect_object === '1'
+		const width = parseFloat(isObjectDetectionSeparate  && d.mon.detector_scale_y_object ? d.mon.detector_scale_y_object : d.mon.detector_scale_y)
+		const height = parseFloat(isObjectDetectionSeparate  && d.mon.detector_scale_x_object ? d.mon.detector_scale_x_object : d.mon.detector_scale_x)
+
+		tx({
+			f:'trigger',
+			id:d.id,
+			ke:d.ke,
+			details:{
+				plug: config.plug,
+				name: `PlateRecognizer`,
+				reason: 'object',
+				matrices: mats,
+				imgHeight: width,
+				imgWidth: height,
+			},
+			frame: frameBuffer
+		})
+	}
+	callback()
+}