Semi-Autonomous-Robot / Semi Autonomous Robot - ECE 4180

Code for controlling mbed hardware (LED's, motors), as well as code for the Raspberry Pi to run a Support Vector Machine that identifies objects using the Pi camera

Dependencies:   mbed Motordriver mbed-rtos PololuLedStrip

collectImages.py@3:a3ed7ff99772, 2019-12-06 (annotated)

Committer:
arogliero3
Date:
Fri Dec 06 00:58:02 2019 -0500
Revision:
3:a3ed7ff99772
Parent:
0:e0dbd261724a 
update img6

Who changed what in which revision?

UserRevisionLine numberNew contents of line
arogliero3 0:e0dbd261724a 1 import sys
arogliero3 0:e0dbd261724a 2 import time
arogliero3 0:e0dbd261724a 3 import picamera
arogliero3 0:e0dbd261724a 4
arogliero3 0:e0dbd261724a 5
arogliero3 0:e0dbd261724a 6
arogliero3 0:e0dbd261724a 7 myargs = sys.argv[1:]
arogliero3 0:e0dbd261724a 8
arogliero3 0:e0dbd261724a 9 camera = picamera.PiCamera()
arogliero3 0:e0dbd261724a 10 camera.resolution = (352, 240)
arogliero3 0:e0dbd261724a 11 camera.color_effects = (128, 128) # turn camera to black and white
arogliero3 0:e0dbd261724a 12 camera.start_preview()
arogliero3 0:e0dbd261724a 13 time.sleep(2)
arogliero3 0:e0dbd261724a 14
arogliero3 0:e0dbd261724a 15
arogliero3 0:e0dbd261724a 16 if len(myargs) <= 1:
arogliero3 0:e0dbd261724a 17 sys.exit("Incorrect arguments")
arogliero3 0:e0dbd261724a 18
arogliero3 0:e0dbd261724a 19 num_images_per_type = int(myargs[0]) # number of images to take of each type of object
arogliero3 0:e0dbd261724a 20
arogliero3 0:e0dbd261724a 21 print("Taking ", num_images_per_type, "images each of ", myargs[1:])
arogliero3 0:e0dbd261724a 22
arogliero3 0:e0dbd261724a 23 for image_type in myargs[1:]:
arogliero3 0:e0dbd261724a 24 for i in range(num_images_per_type):
arogliero3 0:e0dbd261724a 25 filename = image_type + "_" + str(i) + ".bmp"
arogliero3 0:e0dbd261724a 26 camera.capture("images/"+filename, format='bmp', resize=(352, 240))
arogliero3 0:e0dbd261724a 27 print(filename)
arogliero3 0:e0dbd261724a 28 time.sleep(1)