Skip to content
/ ml4pi Public

machine learning modules optimized for a raspberry pi

Notifications You must be signed in to change notification settings

ml4a/ml4pi

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

ml4pi

A tiny framework for doing machine learning on Raspberry Pi.

Setup the Pi

Download the latest version of Raspbian and flash your micro SD card with Etcher

Add blank file called ssh into the root of the SD disk and a file called wpa_supplicant.conf containing the following (replace with your wifi details):

ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev
update_config=1

network={
    ssid="YOUR_WIFI_NETWORK"
    psk="YOUR_WIFI_PASSWORD"
}

In terminal ssh into the pi:

Default password is 'raspberry'. To change password use the passwd command.

Update the pi:

sudo apt-get update && sudo apt-get upgrade

Install nodejs:

sudo apt-get install nodejs npm git-core

Optional: install nettalk for easy file sharing:

sudo apt-get install netatalk

Reboot:

sudo reboot

Install software

Install pre-requisites:

sudo apt install libatlas-base-dev libjasper-dev libqtgui4 libqt4-test libhdf5-dev

Make sure you enable your camera through sudo raspi-config. Reboot again afterwards.

Clone this library:

git clone https://github.com/ml4a/ml4pi

Install all the required python libraries:

cd ml4pi
pip3 install -r requirements.txt 

Try running the interactive trainer:

python3 train_webcam.py

Todo

  • get picamera at faster fps
  • train from a directory of images
  • saving/loading models
  • deployment script (load model, then continuously predict samples)

Training a dataset from a folder of images

(not finished yet)

Example is using a dataset which can be obtained:

wget https://www.vision.caltech.edu/Image_Datasets/Caltech101/101_ObjectCategories.tar.gz
tar -xzf 101_ObjectCategories.tar.gz

About

machine learning modules optimized for a raspberry pi

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published