Skip to main content

AI-Lens-Python Samples Code

Add Python File

We've created a codebase EF_Produce_MicroPython-master for a convenient python programming to AI Lens, you will only need to use the functions and modify the parameters for application.

Download the package and unzip it: EF_Produce_MicroPython-master

Go to Python editor

For programming, we need to add the file of AILens.py. Click Load/Save first and again with Show Files(1), choose "add file" to get to the download page to find the file folder of EF_Produce_MicroPython-master, then add AILens.py.

API

APIDescription
AILENS()Init AI Lens
switch_function(func)Choose AI Lens functions,funcchoose function:(Learncharacteristics learn;Cardcard recognition;Faceface recognition;Trackingtracking recognition;Colorcolor recognition;Ballball recognition)
get_card_content()Recognize the contents on the card
get_card_data()Get the info of the cards from the AI Lens to a list(X-axis; Y-axis; Width; Height; Confidence coefficient; Total numbers of cards; ID of current cards)
get_face()Judge if there is a human face recognized in the AI Lens
get_face_data()Get the info of the face(s) from the AI Lens to a list(X-axis; Y-axis; Width; Height; Confidence coefficient; Total numbers of face(s); ID of current face(s))
get_ball_color()Recognize the color of the balls in the Ai Lens
get_ball_data()Get the info of the ball(s) from the AI Lens to a list(X-axis; Y-axis; Width; Height; Confidence coefficient; Total numbers of ball(s); ID of current ball(s)).
get_track_data()Get the info of the segment(s) from the AI Lens to a list(Deviation angel, deviation distance, segment length)
get_color_type()Recognize the color of the cards in the AI Lens
get_color_data()Get the info of the color(s) from the AI Lens to a list(X-axis; Y-axis; Width; Height; Confidence coefficient; Total numbers of color(s); ID of current color(s)).
learn_object(learn_id)Name with an ID number for a learnt object, ID number: 1~5
get_learn_data()Get the info of the learnt object(s) from the AI Lens to a list(ID of the object(s); Confidence coefficient).

class AILENS(object)

Init AI Lens

def switch_function(self, func)

Choose AI Lens functions, funcchoose function, Learncharacteristics learn, Cardcard recognition;Faceface recognition;Trackingtracking recognition;Colorcolor recognition;Ballball recognition)

def get_image(self)

Get an image

def get_ball_color(self):

Recognize the colors of the balls in the Ai Lens(If ball is blue, the returned result is 'Blue'; if the ball is red, the result is 'Red'; or it returned with a ‘No Ball’ )

def get_ball_data(self)

Return the info of the balls in the AI Lens,BallData [x,y,w,h,confidence,total,order]

x:X-axis

y:Y-axis

w:width

h:height

confidence:Confidence coefficient

total:total numbers of the balls in the AI Lens

order:ID of the current ball

def get_face(self)

Judge if any face(s) in the Ai Lens

def get_face_data(self)

Return the info of human face in the AI lens, FaceData [x,y,w,h,confidence,total,order]

x:X-axis

y:Y-axis

w:width

h:height

confidence:Confidence coefficient

total:total numbers of the face(s) in the AI Lens

order:ID ID of the current face

def get_card_content(self)

Return the contents on the cards, the possible value are:

numberCards = ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]

letterCards = ["A", "B", "C", "D", "E"]

otherCards = ["Mouse", "micro:bit", "Ruler", "Cat", "Peer", "Ship", "Apple", "Car", "Pan", "Dog", "Umbrella", "Airplane", "Clock", "Grape", "Cup", "Turn left", "Turn right", "Forward", "Stop", "Back"]

If no cards get recognized, the returned value is ‘No Card’

def get_color_type(self)

Return the colors of the cards, the possible colors are:

Green card:"Green"

Blue card:"Blue"

Yellow card:"Yellow"

Black card: "Black"

Red card: "Red"

White card: "White"

If no colors get recognized, the returned value is ‘No Color’

def get_color_data(self)

Return the info of the color(s), ColorData [x,y,w,h,confidence,total,order]

x:X-axis

y:Y-axis

w:width

h:height

confidence:Confidence coefficient

total:total numbers of the color(s) in the AI Lens

order: ID of the current color

def get_track_data(self)

Return the info of the segment(s) LineData [angel,width,len]

angel:return the deviation angel of the segment in the AI Lens

width: return the deviation distance of the segment in the AI Lens

len: return the length of the segment in the AI Lens

def learn_object(self, learn_id)

Name with an ID number for a learnt object

ID of the learnt object(s): learn_id

def get_learn_data(self)

Return the info of the learnt object(s), return info: LearnData [ID,confidence]

Object(s) ID:ID

Confidence coefficient of the learnt object:confidence

Samples

Sample 1: Characteristics learn

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Learn)
while True:
ai.get_image()
if button_a.is_pressed():
ai.learn_object(1)
buff = ai.get_learn_data()
if buff[0] == 1:
display.show(Image.HAPPY)
else:
display.show(Image.SAD)


Result

Press button A to learn the object, a smile face displays on the micro:bit while it recognizes the object or it displays a sad face.

Sample 2: Color recognition

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Color)
while True:
ai.get_image()
if (ai.get_color_type() == "Green"):
display.show(Image.HAPPY)
sleep(2000)
else:
display.show(Image.SAD)

Result

A smile face displays on the micro:bit for two seconds if the green card is recognized or it displays a sad face.

Sample 3: Balls recognition

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Ball )
while True:
ai.get_image()
if (ai.get_ball_color() == "Red"):
display.show(Image.HAPPY)
sleep(2000)
else:
display.show(Image.SAD)

Result

A smile face displays on the micro:bit for two seconds if the red ball is recognized or it displays a sad face.

Sample 4: Recognize the black line

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Tracking)
while True:
ai.get_image()
buff = ai.get_track_data()

if buff[1] > 100:
display.show(Image.HAPPY)
else:
display.show(Image.SAD)

Result

The micro:bit displays a sad face if the black line was deteacted deviating to the left by the AI lens; It displays a smile face if the black line was deteacted deviating to the right by the AI lens

Sample 5: Human face(s) recognition

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Face )
while True:
ai.get_image()

if ai.get_face():
display.show(Image.HAPPY)
else:
display.show(Image.SAD)

Result

A smile face displays on the micro:bit if the human face is detected or it displays a sad face.

Sample 6: Cards recognition

from microbit import *
from AILens import *
ai = AILENS()
ai.switch_function(Card)
while True:
ai.get_image()

if (ai.get_card_content() == "Stop"):
display.show(Image.HAPPY)
sleep(2000)
else:
display.show(Image.SAD)

Result

A smile face displays on the micro:bit if a "stop" card is detected or it displays a sad face.