e0f41b7401
Add awb, blc, cproc, filter, and gamma to the tuning file. These don't need any configuration. At the moment there are no inter-module dependencies in the tuning process. We can therefore safely sort them alphabetically. As soon as the first dependency gets introduced (most likely lsc -> ccm) we will see how to solve that. The output order controls the order of processing in the IPA. It is now also in alphabetical order which happens to be no change for the modules that existed previously. For the others, there is no need for a specific order. Signed-off-by: Stefan Klug <stefan.klug@ideasonboard.com> Reviewed-by: Paul Elder <paul.elder@ideasonboard.com> Reviewed-by: Daniel Scally <dan.scally@ideasonboard.com>
62 lines
2.1 KiB
Python
Executable file
62 lines
2.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
#
|
|
# Copyright (C) 2022, Paul Elder <paul.elder@ideasonboard.com>
|
|
#
|
|
# Tuning script for rkisp1
|
|
|
|
import coloredlogs
|
|
import logging
|
|
import sys
|
|
|
|
import libtuning as lt
|
|
from libtuning.parsers import YamlParser
|
|
from libtuning.generators import YamlOutput
|
|
from libtuning.modules.lsc import LSCRkISP1
|
|
from libtuning.modules.agc import AGCRkISP1
|
|
from libtuning.modules.ccm import CCMRkISP1
|
|
from libtuning.modules.static import StaticModule
|
|
|
|
coloredlogs.install(level=logging.INFO, fmt='%(name)s %(levelname)s %(message)s')
|
|
|
|
awb = StaticModule('Awb')
|
|
blc = StaticModule('BlackLevelCorrection')
|
|
color_processing = StaticModule('ColorProcessing')
|
|
filter = StaticModule('Filter')
|
|
gamma_out = StaticModule('GammaOutCorrection', {'gamma': 2.2})
|
|
|
|
tuner = lt.Tuner('RkISP1')
|
|
tuner.add(AGCRkISP1(debug=[lt.Debug.Plot]))
|
|
tuner.add(awb)
|
|
tuner.add(blc)
|
|
tuner.add(CCMRkISP1(debug=[lt.Debug.Plot]))
|
|
tuner.add(color_processing)
|
|
tuner.add(filter)
|
|
tuner.add(gamma_out)
|
|
tuner.add(LSCRkISP1(
|
|
debug=[lt.Debug.Plot],
|
|
# This is for the actual LSC tuning, and is part of the base LSC
|
|
# module. rkisp1's table sector sizes (16x16 programmed as mirrored
|
|
# 8x8) are separate, and is hardcoded in its specific LSC tuning
|
|
# module.
|
|
sector_shape=(17, 17),
|
|
|
|
sector_x_gradient=lt.gradient.Linear(lt.Remainder.DistributeFront),
|
|
sector_y_gradient=lt.gradient.Linear(lt.Remainder.DistributeFront),
|
|
|
|
# This is the function that will be used to average the pixels in
|
|
# each sector. This can also be a custom function.
|
|
sector_average_function=lt.average.Mean(),
|
|
|
|
# This is the function that will be used to smooth the color ratio
|
|
# values. This can also be a custom function.
|
|
smoothing_function=lt.smoothing.MedianBlur(3),
|
|
))
|
|
|
|
tuner.set_input_parser(YamlParser())
|
|
tuner.set_output_formatter(YamlOutput())
|
|
tuner.set_output_order([AGCRkISP1, awb, blc, CCMRkISP1, color_processing,
|
|
filter, gamma_out, LSCRkISP1])
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(tuner.run(sys.argv))
|