Custom Tracking Models

Import KOI 2 Library

from future import *
from koi2 import KOI2

Serial Initialization

koi = KOI2(tx, rx)

Initializes the serial connection to KOI.

Parameters:

tx: TX Pin, use P2 for Robotbit EDU

rx: RX Pin, use P12 for Robotbit EDU

Select Custom Tracking Mode

koi.setModel(3)

Sets the KOI mode to Custom Tracking.

Load Custom Tracking Model

koi.initCustomModel(name, anchor)
koi.initCustomModel('/sd/ballRGB.kmodel',[1.25,1.25,1.50,1.50,1.72,1.72,1.97,1.97,2.34,2.31]) #example for color balls model

Loads a custom model.

Parameters:

name: File name

anchor: A list containing the anchor values.

Get Item Class Coordinates

koi.xywh[0]

Returns coordinates for the detected custom class.

Parameters:

data: The type of data to return.

data	Meaning
0	X coordinates
1	Y coordinates
2	Width of item
3	Height of item

Get Item ID

koi.strVal

Returns the name of the item ID.

Sample Code: Custom Tracking Model

from future import *
from koi2 import KOI2



koi = KOI2('P2', 'P12')
koi.setModel(3)
sleep(15)
koi.initCustomModel(10616832,[1.25,1.25,1.50,1.50,1.72,1.72,1.97,1.97,2.34,2.31])
koi.direction(2)
koi.mirror(0)
screen.sync = 0
while True:
  koi.read_from_uart()
  screen.fill((0, 0, 0))
  screen.text(koi.numberVal,5,10,2,(255, 255, 255))
  screen.text(koi.xywh[0],5,30,2,(255, 255, 255))
  screen.text(koi.xywh[1],5,50,2,(255, 255, 255))
  screen.text(koi.xywh[2],5,70,2,(255, 255, 255))
  screen.text(koi.xywh[3],5,90,2,(255, 255, 255))
  screen.refresh()

556B

Custom Ball Model.py

Open