python 刺激介面初步實現

標籤：port   ==   執行   代碼執行   waitkey   span   show   lse   python   

基於SSVEP的BCI系統需要刺激介面，在PC端用python（主要OpenCV）實現，以下為代碼

import cv2import numpy as npdef up(img,flag):    for i in range(0,100):        for j in range(206,306):            if flag:                img[i][j]=0            else:                img[i][j]=255def down(img,flag):    for i in range(412,512):        for j in range(206,306):            if flag:                img[i][j]=0            else:                img[i][j]=255def left(img,flag):    for i in range(206,306):        for j in range(0,100):            if flag:                img[i][j]=0            else:                img[i][j]=255def right(img,flag):    for i in range(206,306):        for j in range(412,512):            if flag:                img[i][j]=0            else:                img[i][j]=255    if __name__==‘__main__‘:    img=np.zeros((512,512,3),np.uint8)    up(img,0)    down(img,0)    left(img,0)    right(img,0)    cv2.imshow(‘image‘,img)    num=0    while 1:        num+=1        if num>10000000:            num=0        cv2.waitKey(10)        if num%5==0:            left(img,num%2)                if num%6==0:            down(img,num/6%2)        if num%7==0:            right(img,num%2)        if num%8==0:            up(img,num/8%2)        cv2.imshow(‘image‘,img)

效果為：上下左右四個方塊按照一定頻率閃爍，理論上需要6.25,8,9,10，這裡精細度取得比較粗，大概實現功能就OK了，考慮代碼執行時間、機器差異，有需要再更改。

 

python 刺激介面初步實現