26

特定のディレクトリ内のすべての画像でサーフを検索し、それらのキーポイントと記述子を将来の使用のために保存したいと考えています。以下に示すように、ピクルスを使用することにしました。

#!/usr/bin/env python
import os
import pickle
import cv2

class Frame:
  def __init__(self, filename):
    surf = cv2.SURF(500, 4, 2, True)
    self.filename = filename
    self.keypoints, self.descriptors = surf.detect(cv2.imread(filename, cv2.CV_LOAD_IMAGE_GRAYSCALE), None, False)

if __name__ == '__main__':

  Fdb = open('db.dat', 'wb')
  base_path = "img/"
  frame_base = []

  for filename in os.listdir(base_path):
    frame_base.append(Frame(base_path+filename))
    print filename

  pickle.dump(frame_base,Fdb,-1)

  Fdb.close()

実行しようとすると、次のエラーが発生します。

File "src/pickle_test.py", line 23, in <module>
    pickle.dump(frame_base,Fdb,-1)
...
pickle.PicklingError: Can't pickle <type 'cv2.KeyPoint'>: it's not the same object as cv2.KeyPoint

誰が知っていますか、それは何を意味し、どのように修正するのですか? Python 2.6 と Opencv 2.3.1 を使用しています

本当にありがとうございました

4

3 に答える 3

27

問題は、cv2.KeyPoint を pickle ファイルにダンプできないことです。私は同じ問題を抱えていましたが、基本的にキーポイントを自分でシリアル化および逆シリアル化してから、Pickle でダンプすることで回避できました。

したがって、すべてのキーポイントとその記述子をタプルで表します。

temp = (point.pt, point.size, point.angle, point.response, point.octave, 
        point.class_id, desc)

これらすべてのポイントをリストに追加してから、Pickle でダンプします。

次に、データを再度取得する場合は、Pickle を使用してすべてのデータを読み込みます。

temp_feature = cv2.KeyPoint(x=point[0][0],y=point[0][1],_size=point[1], _angle=point[2], 
                            _response=point[3], _octave=point[4], _class_id=point[5]) 
temp_descriptor = point[6]

上記のコードを使用してこのデータから cv2.KeyPoint を作成すると、これらのポイントを使用して特徴のリストを作成できます。

これを行うためのより適切な方法があると思いますが、上記は私にとってはうまく (そして速く) 動作します。私の機能は形式固有のリストに保存されているため、データ形式を少しいじる必要があるかもしれません。私は、一般的なベースで私のアイデアを使用して上記を提示しようとしました。これがお役に立てば幸いです。

于 2012-08-16T10:15:11.820 に答える