svm_video.py 文件源码

def train():
    training_set=[]
    training_labels=[]
    os.chdir("/Users/muyunyan/Desktop/EC500FINAL/logo/")
    counter=0
    a=os.listdir(".")
    for i in a:
     os.chdir(i)
     print(i)
     for d in os.listdir("."):
         img = cv2.imread(d)
         res=cv2.resize(img,(250,250))
         gray_image = cv2.cvtColor(res, cv2.COLOR_BGR2GRAY)
         xarr=np.squeeze(np.array(gray_image).astype(np.float32))
         m,v=cv2.PCACompute(xarr)
         arr= np.array(v)
         flat_arr= arr.ravel()
         training_set.append(flat_arr)
         training_labels.append(i)
     os.chdir("..")
     trainData=training_set
     responses=training_labels
     svm = svm.SVC()
     svm.fit(trainData,responses)
     return svm