/Trash-Sorting-Robot

Trash Sorting Robot using Deep Learning and Robotic Arm

Primary LanguagePythonMIT LicenseMIT

Trash_Sorting_Robot

Helpful Resources:

Repository Contents:

Arduino - files used for Arduino to control robotic arm:

  • basicMovement - .ino file to perform basic movement with robotic arm
  • roboticArm - .ino file to control robotic arm to pick up and drop off a piece of recycling detected from the Raspberry Pi

Documentation - files documenting my build process:

Images - images of my project

RPI - files used for Raspberry Pi to perform object detection:

Tensorflow - files used for training model:

  • CSV - .csv files of both the train and test image set
  • Images - .jpg images of recycling materials and .xml files of the annotated images
  • SaveModel - frozen graph of trained model
  • TFLite - .tflite file of trained model and .tflite file of compiled model for Google Coral USB Accelerator
  • TFRecord - .record files of both the train and test images set used for training
  • ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03 - .config, .ckpt, and .pb files of the quantized model used for transfer learning
  • labelmap.pbtxt - label map used for training

Tutorial:

Hackser.io Tutorial