summaryrefslogtreecommitdiff
path: root/utils/raspberrypi/ctt
diff options
context:
space:
mode:
Diffstat (limited to 'utils/raspberrypi/ctt')
-rw-r--r--utils/raspberrypi/ctt/ctt_macbeth_locator.py8
-rw-r--r--utils/raspberrypi/ctt/ctt_ransac.py14
2 files changed, 10 insertions, 12 deletions
diff --git a/utils/raspberrypi/ctt/ctt_macbeth_locator.py b/utils/raspberrypi/ctt/ctt_macbeth_locator.py
index 55ab4fdb..63dbc4a1 100644
--- a/utils/raspberrypi/ctt/ctt_macbeth_locator.py
+++ b/utils/raspberrypi/ctt/ctt_macbeth_locator.py
@@ -363,9 +363,8 @@ def get_macbeth_chart(img, ref_data):
"""
find Contours in image
"""
- conts, _ = cv2.findContours(edges,
- cv2.RETR_TREE,
- cv2.CHAIN_APPROX_NONE)
+ conts, _ = cv2.findContours(edges, cv2.RETR_TREE,
+ cv2.CHAIN_APPROX_NONE)
if len(conts) == 0:
raise MacbethError(
'\nWARNING: No macbeth chart found!'
@@ -381,8 +380,7 @@ def get_macbeth_chart(img, ref_data):
conts_per = []
for i in range(len(conts)):
per = cv2.arcLength(conts[i], True)
- poly = cv2.approxPolyDP(conts[i],
- epsilon*per, True)
+ poly = cv2.approxPolyDP(conts[i], epsilon*per, True)
if len(poly) == 4 and cv2.isContourConvex(poly):
conts_per.append((poly, per))
diff --git a/utils/raspberrypi/ctt/ctt_ransac.py b/utils/raspberrypi/ctt/ctt_ransac.py
index f7a722e0..a1f5e8c9 100644
--- a/utils/raspberrypi/ctt/ctt_ransac.py
+++ b/utils/raspberrypi/ctt/ctt_ransac.py
@@ -29,11 +29,11 @@ def get_square_verts(c_err=0.05, scale=scale):
square_verts = []
square_0 = np.array(((0, 0), (0, side),
- (side, side), (side, 0)), np.float32)
+ (side, side), (side, 0)), np.float32)
offset_0 = np.array((b_bord_x, b_bord_y), np.float32)
c_off = side * c_err
offset_cont = np.array(((c_off, c_off), (c_off, -c_off),
- (-c_off, -c_off), (-c_off, c_off)), np.float32)
+ (-c_off, -c_off), (-c_off, c_off)), np.float32)
square_0 += offset_0
square_0 += offset_cont
"""
@@ -41,16 +41,16 @@ def get_square_verts(c_err=0.05, scale=scale):
"""
for i in range(6):
shift_i = np.array(((i*side, 0), (i*side, 0),
- (i*side, 0), (i*side, 0)), np.float32)
+ (i*side, 0), (i*side, 0)), np.float32)
shift_bord =np.array(((i*s_bord, 0), (i*s_bord, 0),
- (i*s_bord, 0), (i*s_bord, 0)), np.float32)
+ (i*s_bord, 0), (i*s_bord, 0)), np.float32)
square_i = square_0 + shift_i + shift_bord
for j in range(4):
shift_j = np.array(((0, j*side), (0, j*side),
- (0, j*side), (0, j*side)), np.float32)
+ (0, j*side), (0, j*side)), np.float32)
shift_bord = np.array(((0, j*s_bord),
- (0, j*s_bord), (0, j*s_bord),
- (0, j*s_bord)), np.float32)
+ (0, j*s_bord), (0, j*s_bord),
+ (0, j*s_bord)), np.float32)
square_j = square_i + shift_j + shift_bord
square_verts.append(square_j)
# print('square_verts')