ShinobiSystems
/
Shinobi
mirror of https://gitlab.com/Shinobi-Systems/Shinobi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update INSTALL.sh for Tensorflow plugin

pushover
Moe 2020-01-17 19:44:10 -08:00
parent db6d6f90a3
commit a09a47e762
1 changed files with 8 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
plugins/tensorflow/INSTALL.sh
View File

@ -1,20 +1,11 @@
#!/bin/bash #!/bin/bash
mkdir data echo "Shinobi - Do you want to install tensorflowjs with GPU support? "
mkdir data/inception echo "(y)es or (N)o"
chmod -R 777 data read nodejsinstall
wget https://cdn.shinobi.video/weights/inception5h.zip -O inception5h.zip if [ "$nodejsinstall" = "y" ] || [ "$nodejsinstall" = "Y" ]; then
unzip inception5h.zip -d data/inception npm install @tensorflow/tfjs-node-gpu --unsafe-perm
if [ $(dpkg-query -W -f='${Status}' opencv_version 2>/dev/null | grep -c "ok installed") -eq 0 ]; then
echo "Shinobi - Do ypu want to let the `opencv4nodejs` npm package install OpenCV? "
echo "Only do this if you do not have OpenCV already or will not use a GPU (Hardware Acceleration)."
echo "(y)es or (N)o"
read nodejsinstall
if [ "$nodejsinstall" = "y" ] || [ "$nodejsinstall" = "Y" ]; then
export OPENCV4NODEJS_DISABLE_AUTOBUILD=0
else
export OPENCV4NODEJS_DISABLE_AUTOBUILD=1
fi
else else
export OPENCV4NODEJS_DISABLE_AUTOBUILD=1 npm install @tensorflow/tfjs-node --unsafe-perm
fi fi
npm install opencv4nodejs moment express canvas@1.6 --unsafe-perm npm install @tensorflow-models/coco-ssd --unsafe-perm
npm install buffer-to-uint8array