pavel/kick-it
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Numpy 1.6.0 compatibility changes

Browse Source
This commit is contained in:
Pavel Lutskov
2018-06-24 14:06:47 +02:00
parent 60dc7d477e
commit a28516177d
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pykick/finders.py
View File

@@ -87,12 +87,9 @@ class GoalFinder(object):
thr = self.primary_mask(frame) thr = self.primary_mask(frame)
cnts, _ = cv2.findContours(thr, cv2.RETR_EXTERNAL, cnts, _ = cv2.findContours(thr, cv2.RETR_EXTERNAL,
cv2.CHAIN_APPROX_SIMPLE) cv2.CHAIN_APPROX_SIMPLE)
areas = np.array([cv2.contourArea(cnt) for cnt in cnts]) cnts.sort(key=cv2.contourArea, reverse=True)
# Candidates are at most 6 biggest white areas
top_x = 6 top_x = 6
if len(areas) > top_x: cnts = cnts[:top_x]
cnt_ind = np.argpartition(areas, -top_x)[-top_x:]
cnts = [cnts[i] for i in cnt_ind]
epsilon = [0.01 * cv2.arcLength(cnt, True) for cnt in cnts] epsilon = [0.01 * cv2.arcLength(cnt, True) for cnt in cnts]