setup.sh

#!/bin/bash
# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This script is designed to run on the Raspberry Pi OS.
# It verifies the camera is enabled and installs Coral libraries.

set -e

function enable_camera () {
    CAM=$(sudo raspi-config nonint get_camera)
    if [ $CAM -eq 1 ]; then
        sudo raspi-config nonint do_camera 0
        echo "Camera is now enabled, but you must reboot for it to take effect."
        echo "After reboot, run setup.sh again to finish the setup."

        while true; do
          read -p "Reboot now? (y/n) " yn
          case $yn in
            [Yy]* )
              echo "Rebooting..."
              sudo reboot now; break;;
            [Nn]* )
              echo "Setup cancelled. You must reboot to continue setup."; exit;;
            * )
              echo "Please answer yes or no.";;
          esac
        done
    else
      echo "Camera is already enabled."
    fi
}

echo
echo "Checking the camera..."
enable_camera

echo
echo "Installing required packages for Coral..."
echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" | sudo tee /etc/apt/sources.list.d/coral-edgetpu.list
curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key add -

sudo apt-get update --allow-releaseinfo-change

sudo apt-get -y install \
  libedgetpu1-max \
  python3-pycoral \
  python3-tflite-runtime \
  python3-pyaudio \
  python3-opencv \
  libatlas-base-dev \
  zip \
  unzip

python3 -m pip install pip --upgrade
python3 -m pip install \
  pynput \
  tflite-support

git clone https://github.com/google-coral/aiy-maker-kit
python3 -m pip install ./aiy-maker-kit
bash aiy-maker-kit/examples/download_models.sh
bash aiy-maker-kit/projects/download_models.sh

echo
echo "Coral software setup is complete."
echo "To continue setup on a Raspberry Pi, follow the guide at g.co/aiy/maker"
echo