編集履歴

回答編集履歴

追記

2018/01/11 15:51

投稿

スコア35660

test CHANGED Viewed

@@ -268,7 +268,7 @@
-**partition.py**
+**partition.py** [OpenCV内部実装](https://github.com/opencv/opencv/blob/c7049ca627f47fece7adb2c132ba118209d517e6/modules/core/include/opencv2/core/operations.hpp#L440)の焼き直し
 ```Python

説明追記

2018/01/11 15:51

投稿

LouiS0616

スコア35660

test CHANGED Viewed

@@ -148,6 +148,8 @@
 その部分さえ修正できれば、充分実用できるかと。
 **find_buoy.py**
 ```Python
@@ -168,6 +170,8 @@
+# 質問のブイマスク抽出の流用
 def make_buoy_mask(img):
     return cv2.inRange(
@@ -182,6 +186,8 @@
+# マスク画像を生成してから、ブイの座標のnp.arrayを返す
 def find_buoy(img):
     return np.vstack(
@@ -272,6 +278,8 @@
+# 番号を振りなおす  e.g. [0, 0, 3, 5, 3, 2] -> [0, 0, 2, 3, 2, 1]
 def renumber(id_list):
     id_set = set(id_list)
@@ -288,6 +296,10 @@
+# UnionFind法を用いて、データを分類していく
+# 同一のクラスタであるとの判断基準はpredicate_funcで与える
 def partition(src_list, predicate_func):
     src_len = len(src_list)

追記

2018/01/11 15:48

投稿

LouiS0616

スコア35660

test CHANGED Viewed

@@ -137,3 +137,231 @@
 **実際**
 ![実際](46dea120c1c96f487f5d957307d1bae6.jpeg)
+『ブイ候補』を正しく認識できたなら...
+---
+前述の理由により正しく動作しませんが... (make_buoy_mask関数に相当)
+その部分さえ修正できれば、充分実用できるかと。
+**find_buoy.py**
+```Python
+import sys
+import numpy as np
+import cv2
+from partition import partition
+def make_buoy_mask(img):
+    return cv2.inRange(
+        cv2.cvtColor(img, cv2.COLOR_BGR2HSV),
+        lowerb=np.array([170,100,190]),
+        upperb=np.array([200,255,255])
+    )
+def find_buoy(img):
+    return np.vstack(
+        np.where(make_buoy_mask(img)==255)
+    ).T
+def main(img_file):
+    img = cv2.imread(img_file)
+    # Find buoy pixels
+    buoy_pts = find_buoy(img)
+    buoy_ids = partition(
+        buoy_pts,
+        lambda x, y: np.linalg.norm(x-y) <= 10
+    )
+    # Get center of buoys
+    num_of_buoy = max(buoy_ids) + 1
+    typical_buoy_pts = [
+        buoy_pts[buoy_ids.index(i)]
+        for i in range(num_of_buoy)
+    ]
+    # Check buoy position
+    RED = (0, 0, 255)
+    for buoy in typical_buoy_pts:
+        cv2.circle(img, tuple(buoy), radius=10, color=RED, thickness=5)
+    # Show result
+    cv2.imshow('result', img)
+    cv2.waitKey()
+    cv2.destroyAllWindows()
+if __name__ == '__main__':
+    if len(sys.argv) != 2:
+        print('Usage: find_buoy.py img_file')
+        exit()
+    main(img_file=sys.argv[1])
+```
+**partition.py**
+```Python
+from itertools import product
+from UnionFind import UnionFind
+def renumber(id_list):
+    id_set = set(id_list)
+    replace_dict = dict(zip(
+        sorted(list(id_set)),
+        [i for i, _ in enumerate(id_set)]
+    ))
+    return [replace_dict[elem] for elem in id_list]
+def partition(src_list, predicate_func):
+    src_len = len(src_list)
+    uf = UnionFind(src_len)
+    loop_obj = product(src_list, src_list)
+    for ij, (ei, ej) in enumerate(loop_obj):
+        i, j = divmod(ij, src_len)
+        if ei is ej:
+            continue
+        if not predicate_func(ei, ej):
+            continue
+        uf.union(i, j)
+    return renumber(uf._id)
+```
+**test_partition.py**
+```Python
+import unittest
+from partition import partition
+class TestPartition(unittest.TestCase):
+    def test_partition(self):
+        self.assertEqual(
+            partition([1, 2, 3, 7, 8, 9, 2], lambda x, y: abs(x-y) <= 1),
+            [0, 0, 0, 1, 1, 1, 0]
+        )
+        self.assertEqual(
+            partition([1, 1, 1, 3, 4, 4, 5, 6], lambda x, y: x % 2 == y % 2),
+            [0, 0, 0, 0, 1, 1, 0, 1]
+        )
+if __name__ == '__main__':
+    unittest.main()
+```
+UnionFind.pyは、[既にあるもの](https://gist.github.com/SonechkaGodovykh/18f60a3b9b3e6812c071456f61f9c5a6)を利用。

追記

2018/01/11 15:42

投稿

LouiS0616

スコア35660

test CHANGED Viewed

@@ -55,3 +55,85 @@
 適当にテスト画像を用意してくださると回答しやすいので、次回以降は気にかけてみてください。
+そもそもブイを正しく抽出できてなさそう
+---
+次のコードで実験しました、が...
+```Python
+import cv2
+import numpy as np
+def main():
+    img = cv2.imread('sea.jpg')
+    hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+    lower_red = np.array([170,100,190])
+    upper_red = np.array([200,255,255])
+    mask = cv2.inRange(hsv_img, lower_red, upper_red)
+    #
+    # Write points
+    points = np.vstack(
+        np.where(mask==255)
+    ).T
+    RED = (0, 0, 255)
+    for pt in map(tuple, points):
+        cv2.circle(img, pt, radius=20, color=RED, thickness=-1)
+    cv2.imwrite('result.jpg', img)
+    #
+    #
+if __name__ == "__main__":
+    main()
+```
+**期待**
+![期待](714fa0fa145802b955c0111eb97ab977.jpeg)
+---
+**実際**
+![実際](46dea120c1c96f487f5d957307d1bae6.jpeg)