initial commit

Author: AIWintermuteAI

color_detection.py

@@ -0,0 +1,35 @@
+import video, sensor, image, lcd, time
+
+sensor.reset()
+sensor.set_pixformat(sensor.RGB565)
+sensor.set_framesize(sensor.QVGA)
+sensor.run(1)
+sensor.skip_frames(30)
+sensor.set_auto_gain(False) # must be turned off for color tracking
+sensor.set_auto_whitebal(False) # must be turned off for color tracking
+
+v = video.open("/sd/capture.avi", record=1, interval=200000, quality=50)
+
+def get_average(histogram):
+	sum = 0
+	average = 0
+	for object in histogram:
+	   sum = average + object
+	average = sum / 8
+	return average 
+
+r = [(320//2)-(50//2), (240//2)-(50//2), 50, 50] # 50x50 center of QVGA.
+tim = time.ticks_ms()
+
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    img = sensor.snapshot()
+    img.draw_rectangle(r)
+    hist = img.get_statistics(bins=8,roi=r)
+    rgb_value = image.lab_to_rgb((hist.l_mean(),hist.a_mean(),hist.b_mean()))
+    img.draw_string(0, 0, str(rgb_value), color = (rgb_value[0], rgb_value[1], rgb_value[2]), scale = 2)
+    img_len = v.record(img)
+    lcd.display(img)
+print("finish")
+v.record_finish()
+lcd.clear()
+

detect_eyes.py

@@ -0,0 +1,35 @@
+import video, sensor, image, lcd, time
+
+lcd.init()
+# Reset sensor
+sensor.reset()
+
+# Sensor settings
+sensor.set_contrast(1)
+sensor.set_gainceiling(16)
+sensor.set_framesize(sensor.QVGA)
+sensor.set_pixformat(sensor.RGB565)
+sensor.run(1)
+sensor.skip_frames(30)
+v = video.open("/sd/capture_eyes.avi", record=1, interval=200000, quality=50)
+
+# Load Haar Cascade
+# By default this will use all stages, lower satges is faster but less accurate.
+face_cascade = image.HaarCascade("frontalface", stages=25)
+eyes_cascade = image.HaarCascade("eye", stages=24)
+tim = time.ticks_ms()
+
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    # Capture snapshot
+    img = sensor.snapshot()
+    objects = img.find_features(face_cascade, threshold=0.5, scale_factor=1.5)
+    for face in objects:
+        img.draw_rectangle(face)
+        eyes = img.find_features(eyes_cascade, threshold=0.5, scale_factor=1.2, roi=face)
+        for e in eyes:
+            img.draw_rectangle(e)
+    img_len = v.record(img)
+    lcd.display(img)
+print("finish")
+v.record_finish()
+lcd.clear()

find non_inf_lines.py

@@ -0,0 +1,19 @@
+#find non-infinite lines
+import video, sensor, image, lcd, time
+lcd.init()
+sensor.reset()
+sensor.set_pixformat(sensor.RGB565) # grayscale is faster
+sensor.set_framesize(sensor.QVGA)
+sensor.run(1)
+sensor.skip_frames(30)
+v = video.open("/sd/capture_lines.avi", record=1, interval=200000, quality=50)
+tim = time.ticks_ms()
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    img = sensor.snapshot()
+    for l in img.find_line_segments(merge_distance = 10, max_theta_diff = 25, roi=(80,60,160,120)):
+        img.draw_line(l.line(), color = (255, 0, 0))
+    lcd.display(img)
+    img_len = v.record(img)
+print("finish")
+v.record_finish()
+lcd.clear()

find_circles.py

@@ -0,0 +1,26 @@
+#find circles
+import video, sensor, image, lcd, time
+
+lcd.init()
+sensor.reset()
+sensor.set_contrast(1)
+sensor.set_gainceiling(16)
+sensor.set_pixformat(sensor.RGB565)
+sensor.set_framesize(sensor.QVGA)
+sensor.run(1)
+sensor.skip_frames(30)
+v = video.open("/sd/capture_circles.avi", record=1, interval=200000, quality=50)
+
+tim = time.ticks_ms()
+
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    img = sensor.snapshot()
+    for c in img.find_circles(threshold = 3500, x_margin = 10, y_margin = 10, r_margin = 10, r_min = 2, r_max = 100, r_step = 2,roi=(80,60,160,120)):
+        img.draw_circle(c.x(), c.y(), c.r(), color = (255, 0, 0))
+    img.draw_rectangle(80,60,160,120)
+    img_len = v.record(img)
+    lcd.display(img)
+print("finish")
+v.record_finish()
+lcd.clear()
+

find_inf_lines.py

@@ -0,0 +1,26 @@
+#find infinite lines
+import video, sensor, image, lcd, time
+enable_lens_corr = True
+lcd.init()
+sensor.reset()
+sensor.set_pixformat(sensor.RGB565)
+sensor.set_framesize(sensor.QVGA)
+sensor.run(1)
+sensor.skip_frames(time = 2000)
+
+v = video.open("/sd/capture_inf_lines.avi", record=1, interval=200000, quality=50)
+min_degree = 0
+max_degree = 179
+tim = time.ticks_ms()
+
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    img = sensor.snapshot()
+    for l in img.find_lines(threshold = 4000, theta_margin = 25, rho_margin = 25):
+        if (min_degree <= l.theta()) and (l.theta() <= max_degree):
+            img.draw_line(l.line(), color = (255, 0, 0))
+            print(l)	
+    lcd.display(img)
+    img_len = v.record(img)
+print("finish")
+v.record_finish()
+lcd.clear()

find_rectangles.py

@@ -0,0 +1,25 @@
+#find rects
+import video, sensor, image, lcd, time
+
+lcd.init()
+sensor.reset()
+sensor.set_contrast(1)
+sensor.set_gainceiling(16)
+sensor.set_pixformat(sensor.RGB565)
+sensor.set_framesize(sensor.QVGA)
+sensor.run(1)
+sensor.skip_frames(30)
+v = video.open("/sd/capture_rects.avi", record=1, interval=200000, quality=50)
+tim = time.ticks_ms()
+
+while(time.ticks_diff(time.ticks_ms(), tim)<30000):
+    img = sensor.snapshot()
+    for r in img.find_rects(threshold = 18000,roi=(80,60,160,120)):
+        img.draw_rectangle(r.rect(), color = (255, 0, 0))
+        for p in r.corners(): img.draw_circle(p[0], p[1], 5, color = (0, 255, 0))
+    img.draw_rectangle(80,60,160,120)
+    img_len = v.record(img)
+    lcd.display(img)
+print("finish")
+v.record_finish()
+lcd.clear()

kflash

@@ -0,0 +1 @@
+Subproject commit cb9b5fb282ad1fae1f6e8d5cea5a3dd1bee623c9