An attempt at getting image data back
This commit is contained in:
@@ -0,0 +1,6 @@
|
||||
# SPDX-License-Identifier: GPL-2.0-or-later
|
||||
#
|
||||
# Copyright (C) 2024, Paul Elder <paul.elder@ideasonboard.com>
|
||||
|
||||
from libtuning.modules.ccm.ccm import CCM
|
||||
from libtuning.modules.ccm.rkisp1 import CCMRkISP1
|
||||
@@ -0,0 +1,41 @@
|
||||
# SPDX-License-Identifier: GPL-2.0-or-later
|
||||
#
|
||||
# Copyright (C) 2024, Paul Elder <paul.elder@ideasonboard.com>
|
||||
# Copyright (C) 2024, Ideas on Board
|
||||
#
|
||||
# Base Ccm tuning module
|
||||
|
||||
from ..module import Module
|
||||
|
||||
from libtuning.ctt_ccm import ccm
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class CCM(Module):
|
||||
type = 'ccm'
|
||||
hr_name = 'CCM (Base)'
|
||||
out_name = 'GenericCCM'
|
||||
|
||||
def __init__(self, debug: list):
|
||||
super().__init__()
|
||||
|
||||
self.debug = debug
|
||||
|
||||
def do_calibration(self, images):
|
||||
logger.info('Starting CCM calibration')
|
||||
|
||||
imgs = [img for img in images if img.macbeth is not None]
|
||||
|
||||
# todo: Take LSC calibration results into account.
|
||||
cal_cr_list = None
|
||||
cal_cb_list = None
|
||||
|
||||
try:
|
||||
ccms = ccm(imgs, cal_cr_list, cal_cb_list)
|
||||
except ArithmeticError:
|
||||
logger.error('CCM calibration failed')
|
||||
return None
|
||||
|
||||
return ccms
|
||||
@@ -0,0 +1,28 @@
|
||||
# SPDX-License-Identifier: GPL-2.0-or-later
|
||||
#
|
||||
# Copyright (C) 2024, Paul Elder <paul.elder@ideasonboard.com>
|
||||
# Copyright (C) 2024, Ideas on Board
|
||||
#
|
||||
# Ccm module for tuning rkisp1
|
||||
|
||||
from .ccm import CCM
|
||||
|
||||
|
||||
class CCMRkISP1(CCM):
|
||||
hr_name = 'Crosstalk Correction (RkISP1)'
|
||||
out_name = 'Ccm'
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
# We don't need anything from the config file.
|
||||
def validate_config(self, config: dict) -> bool:
|
||||
return True
|
||||
|
||||
def process(self, config: dict, images: list, outputs: dict) -> dict:
|
||||
output = {}
|
||||
|
||||
ccms = self.do_calibration(images)
|
||||
output['ccms'] = ccms
|
||||
|
||||
return output
|
||||
Reference in New Issue
Block a user