Python Bag of Words Clustering

Question

Edit: Hier ist der vollständige Code für jedermann mit Fragen noch github.com

Ich versuche, eine Bilderkennung Projekt zu tun mit SIFT und Bogen. Bis jetzt versuche ich, mein Wörterbuch zu trainieren und zu bauen. Ich habe Bilder aus 5 verschiedenen Klassen gelesen, berechnete Deskriptoren und fügte sie alle einer Python-Liste ([]) Seite an Seite hinzu. Jetzt versuche ich die Python-Version von BOWMeansTrainer zu verwenden, um meine Deskriptoren mit k = 5 zu clustern (ist das korrekt? Für 5 Klassen?). Ich versuche, Cluster() meinen Descriptor Vektor passieren, aber ich erhalte die Störung

Traceback (most recent call last): 


File "C:\Python27\Project2\beginning.py", line 40, in <module> 
    bow.cluster(des) 
TypeError: descriptors data type = 17 is not supported 

Ich bin nicht wirklich sicher, welches Format in meinem numpy Array zu setzen, hat jemand eine Idee?

sift = cv2.SIFT() 

descriptors = [] 
for path in training_paths: 
    image = cv2.imread(path) 
    print path 
    gray = cv2.cvtColor(image, cv2.CV_LOAD_IMAGE_GRAYSCALE) 
    kp, dsc= sift.detectAndCompute(gray, None) 
    descriptors.append(dsc) 

des = np.array(descriptors) 

k=5 
bow = cv2.BOWKMeansTrainer(k) 
bow.cluster(des) 

Wie Sie halte ich sehen kann, die Sift Deskriptoren anhängt, und dann versuchen, zu einem numpy Array (das benötigte Format) zu konvertieren.

Answer 1

dachte es nur dank der opencv Foren aus, stattdessen eine andere Liste der Verwendung von (I-Deskriptoren oben verwendet), um nur die Deskriptoren fügen Sie direkt in Ihre Tasche mit bow.add (dsc) finden

dictionarySize = 5 

BOW = cv2.BOWKMeansTrainer(dictionarySize) 

for p in training_paths: 
    image = cv2.imread(p) 
    gray = cv2.cvtColor(image, cv2.CV_LOAD_IMAGE_GRAYSCALE) 
    kp, dsc= sift.detectAndCompute(gray, None) 
    BOW.add(dsc) 

#dictionary created 
dictionary = BOW.cluster() 

EDIT: Für alle anderen Probleme habe ich den Rest des Skripts hochgeladen here