From ff90c355b3addbe613575e96ae6177232fd4dbc3 Mon Sep 17 00:00:00 2001 From: Moe Date: Fri, 17 Jan 2020 19:59:46 -0800 Subject: [PATCH] Update INSTALL.sh for Tensorflow plugin+ --- plugins/tensorflow/INSTALL.sh | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/plugins/tensorflow/INSTALL.sh b/plugins/tensorflow/INSTALL.sh index da69b0e3..5c6b510f 100644 --- a/plugins/tensorflow/INSTALL.sh +++ b/plugins/tensorflow/INSTALL.sh @@ -2,10 +2,19 @@ echo "Shinobi - Do you want to install tensorflowjs with GPU support? " echo "(y)es or (N)o" read nodejsinstall +echo "Getting Tensorflow Node.js module..." if [ "$nodejsinstall" = "y" ] || [ "$nodejsinstall" = "Y" ]; then npm install @tensorflow/tfjs-node-gpu --unsafe-perm else npm install @tensorflow/tfjs-node --unsafe-perm fi +echo "Getting Coco SSD Model..." npm install @tensorflow-models/coco-ssd --unsafe-perm +echo "Getting other Libraries..." npm install buffer-to-uint8array +if [ ! -e "./conf.json" ]; then + echo "Creating conf.json" + sudo cp conf.sample.json conf.json +else + echo "conf.json already exists..." +fi