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@0:e0dbd261724a, 2019-12-05 (annotated)
- Committer:
- arogliero3
- Date:
- Thu Dec 05 20:34:10 2019 -0500
- Revision:
- 0:e0dbd261724a
Adding code to mbed repo
Who changed what in which revision?
User | Revision | Line number | New 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) |