libtuning: Fix imports

Fix imports to match new structure in the files copied from Raspberry
Pi. Add missing imports in macbeth.py. Add missing dependencies to
requirements.txt.

Signed-off-by: Stefan Klug <stefan.klug@ideasonboard.com>
Reviewed-by: Paul Elder <paul.elder@ideasonboard.com>
Reviewed-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
Reviewed-by: Daniel Scally <dan.scally@ideasonboard.com>
This commit is contained in:
Stefan Klug 2024-06-11 08:36:43 +02:00
parent 388fe3bcf9
commit b1f3b3f08d
4 changed files with 18 additions and 6 deletions

View file

@ -4,10 +4,12 @@
# #
# camera tuning tool for AWB # camera tuning tool for AWB
from ctt_image_load import *
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from bisect import bisect_left from bisect import bisect_left
from scipy.optimize import fmin from scipy.optimize import fmin
import numpy as np
from .image import Image
""" """

View file

@ -4,12 +4,14 @@
# #
# camera tuning tool for CCM (colour correction matrix) # camera tuning tool for CCM (colour correction matrix)
from ctt_image_load import *
from ctt_awb import get_alsc_patches
import colors
from scipy.optimize import minimize
from ctt_visualise import visualise_macbeth_chart
import numpy as np import numpy as np
from scipy.optimize import minimize
from . import ctt_colors as colors
from .image import Image
from .ctt_awb import get_alsc_patches
from .utils import visualise_macbeth_chart
""" """
takes 8-bit macbeth chart values, degammas and returns 16 bit takes 8-bit macbeth chart values, degammas and returns 16 bit
""" """

View file

@ -1,6 +1,7 @@
# SPDX-License-Identifier: BSD-2-Clause # SPDX-License-Identifier: BSD-2-Clause
# #
# Copyright (C) 2019, Raspberry Pi Ltd # Copyright (C) 2019, Raspberry Pi Ltd
# Copyright (C) 2024, Ideas on Board Oy
# #
# Locate and extract Macbeth charts from images # Locate and extract Macbeth charts from images
# (Copied from: ctt_macbeth_locator.py) # (Copied from: ctt_macbeth_locator.py)
@ -11,6 +12,10 @@ import cv2
import os import os
from pathlib import Path from pathlib import Path
import numpy as np 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 from libtuning.image import Image

View file

@ -1,5 +1,8 @@
matplotlib
numpy numpy
opencv-python opencv-python
py3exiv2 py3exiv2
pyyaml pyyaml
rawpy rawpy
scikit-learn
scipy