summaryrefslogtreecommitdiff
path: root/utils/tuning/libtuning/macbeth.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils/tuning/libtuning/macbeth.py')
-rw-r--r--utils/tuning/libtuning/macbeth.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/utils/tuning/libtuning/macbeth.py b/utils/tuning/libtuning/macbeth.py
index 81f3e87c..265a33d6 100644
--- a/utils/tuning/libtuning/macbeth.py
+++ b/utils/tuning/libtuning/macbeth.py
@@ -1,6 +1,7 @@
# SPDX-License-Identifier: BSD-2-Clause
#
# Copyright (C) 2019, Raspberry Pi Ltd
+# Copyright (C) 2024, Ideas on Board Oy
#
# Locate and extract Macbeth charts from images
# (Copied from: ctt_macbeth_locator.py)
@@ -11,6 +12,10 @@ import cv2
import os
from pathlib import Path
import numpy as np
+import warnings
+from sklearn import cluster as cluster
+
+from .ctt_ransac import get_square_verts, get_square_centres
from libtuning.image import Image