Beispiel #1
0
# Matplot Lib example plotting
plotc = {'r': 'r', 'g': 'g', 'b': 'b', 'i': 'm', 'h': 'y'}
for key in list(tct.data.keys()):
    plt.plot(tct.data[key], plotc[key])
    for pt in tct.peaks[key]:
        plt.plot(pt[0], pt[1], 'r*')
    for pt in tct.valleys[key]:
        plt.plot(pt[0], pt[1], 'b*')
    plt.grid()
plt.show()

disp = Display((800, 600))
while disp.isNotDone():
    img = cam.getImage()
    result = tct.recognize(img)
    plt.plot(tct._rtData, 'r-')
    plt.grid()
    plt.savefig('temp.png')
    plt.clf()
    plotImg = Image('temp.png')

    roi = ROI(img.width * 0.45, img.height * 0.45, img.width * 0.1,
              img.height * 0.1, img)
    roi.draw(width=3)
    img.drawText(str(result), 20, 20, color=Color.RED, fontsize=32)
    img = img.applyLayers()
    img = img.blit(plotImg.resize(w=img.width, h=img.height),
                   pos=(0, 0),
                   alpha=0.5)
    img.save(disp)
roi = ROI(img.width*0.45,img.height*0.45,img.width*0.1,img.height*0.1,img)
tct.train(cam,roi=roi,maxFrames=250,pkWndw=20)

# Matplot Lib example plotting
plotc = {'r':'r','g':'g','b':'b','i':'m','h':'y'}
for key in tct.data.keys():
    plt.plot(tct.data[key],plotc[key])
    for pt in tct.peaks[key]:
        plt.plot(pt[0],pt[1],'r*')
    for pt in tct.valleys[key]:
        plt.plot(pt[0],pt[1],'b*')
    plt.grid()
plt.show()

disp = Display((800,600))
while disp.isNotDone():
    img = cam.getImage()
    result = tct.recognize(img)
    plt.plot(tct._rtData,'r-')
    plt.grid()
    plt.savefig('temp.png')
    plt.clf()
    plotImg = Image('temp.png')    
    
    roi = ROI(img.width*0.45,img.height*0.45,img.width*0.1,img.height*0.1,img)
    roi.draw(width=3)
    img.drawText(str(result),20,20,color=Color.RED,fontsize=32)
    img = img.applyLayers()
    img = img.blit(plotImg.resize(w=img.width,h=img.height),pos=(0,0),alpha=0.5)
    img.save(disp)