NekoCWD/libcamera
NekoCWD/libcamera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 7 Wiki Activity Actions

utils: tuning: rkisp1: Add lux module

Browse source 
Now that the lux module is available, add it to the rkisp1 tuner.

While at it, sort the imports correctly.

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>
This commit is contained in:
Stefan Klug 2025-01-23 12:41:03 +01:00
parent a783a90dec
commit 92bb16b68e
1 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
utils/tuning/rkisp1.py
View file

@ -6,18 +6,19 @@
# #
# Tuning script for rkisp1 # Tuning script for rkisp1
import coloredlogs
import logging import logging
import sys import sys
import coloredlogs
import libtuning as lt import libtuning as lt
from libtuning.parsers import YamlParser
from libtuning.generators import YamlOutput from libtuning.generators import YamlOutput
from libtuning.modules.lsc import LSCRkISP1
from libtuning.modules.agc import AGCRkISP1 from libtuning.modules.agc import AGCRkISP1
from libtuning.modules.awb import AWBRkISP1 from libtuning.modules.awb import AWBRkISP1
from libtuning.modules.ccm import CCMRkISP1 from libtuning.modules.ccm import CCMRkISP1
from libtuning.modules.lsc import LSCRkISP1
from libtuning.modules.lux import LuxRkISP1
from libtuning.modules.static import StaticModule from libtuning.modules.static import StaticModule
from libtuning.parsers import YamlParser
coloredlogs.install(level=logging.INFO, fmt='%(name)s %(levelname)s %(message)s') coloredlogs.install(level=logging.INFO, fmt='%(name)s %(levelname)s %(message)s')
@ -45,12 +46,15 @@ lsc = LSCRkISP1(debug=[lt.Debug.Plot],
# This is the function that will be used to smooth the color ratio # This is the function that will be used to smooth the color ratio
# values. This can also be a custom function. # values. This can also be a custom function.
smoothing_function=lt.smoothing.MedianBlur(3),) smoothing_function=lt.smoothing.MedianBlur(3),)
lux = LuxRkISP1(debug=[lt.Debug.Plot])
tuner = lt.Tuner('RkISP1') tuner = lt.Tuner('RkISP1')
tuner.add([agc, awb, blc, ccm, color_processing, filter, gamma_out, lsc]) tuner.add([agc, awb, blc, ccm, color_processing, filter, gamma_out, lsc, lux])
tuner.set_input_parser(YamlParser()) tuner.set_input_parser(YamlParser())
tuner.set_output_formatter(YamlOutput()) tuner.set_output_formatter(YamlOutput())
tuner.set_output_order([agc, awb, blc, ccm, color_processing,
# Bayesian AWB uses the lux value, so insert the lux algorithm before AWB.
tuner.set_output_order([agc, lux, awb, blc, ccm, color_processing,
filter, gamma_out, lsc]) filter, gamma_out, lsc])
if __name__ == '__main__': if __name__ == '__main__':