[v5,0/4] Raspberry Pi AWB using neural networks
mbox series

Message ID 20260127120604.6560-1-david.plowman@raspberrypi.com
Headers show
Series
  • Raspberry Pi AWB using neural networks
Related show

Message

David Plowman Jan. 27, 2026, 11:59 a.m. UTC 
Hi everyone

Hopefully a fifth and final version!

The only change here is that there is an extra comment at the top of
awb_nn.cpp giving instructions on where to find the model definitions
and source code, the full datasets used for training the models, and
instructions on how to capture your own images and re-train
everything.

Thanks
David

Peter Bailey (4):
  ipa: rpi: controller: awb: Separate Bayesian AWB into AwbBayes
  ipa: rpi: controller: awb: Add Neural Network AWB
  ipa: rpi: controller: Ignore algorithms that are not enabled
  ipa: rpi: pisp: vc4: Update tuning files for new AWB

 meson_options.txt                            |   5 +
 src/ipa/rpi/controller/controller.cpp        |  18 +
 src/ipa/rpi/controller/meson.build           |  10 +
 src/ipa/rpi/controller/rpi/awb.cpp           | 409 ++---------------
 src/ipa/rpi/controller/rpi/awb.h             |  99 ++--
 src/ipa/rpi/controller/rpi/awb_bayes.cpp     | 444 ++++++++++++++++++
 src/ipa/rpi/controller/rpi/awb_nn.cpp        | 456 +++++++++++++++++++
 src/ipa/rpi/pisp/data/imx219.json            |  65 +++
 src/ipa/rpi/pisp/data/imx296.json            |  64 +++
 src/ipa/rpi/pisp/data/imx296_16mm.json       |  64 +++
 src/ipa/rpi/pisp/data/imx296_6mm.json        |  64 +++
 src/ipa/rpi/pisp/data/imx477.json            |  65 +++
 src/ipa/rpi/pisp/data/imx477_16mm.json       |  65 +++
 src/ipa/rpi/pisp/data/imx477_6mm.json        |  65 +++
 src/ipa/rpi/pisp/data/imx477_scientific.json |  79 ++++
 src/ipa/rpi/pisp/data/imx708.json            |  64 +++
 src/ipa/rpi/pisp/data/imx708_wide.json       |  64 +++
 src/ipa/rpi/pisp/data/ov5647.json            |  65 +++
 src/ipa/rpi/vc4/data/imx219.json             |  66 +++
 src/ipa/rpi/vc4/data/imx296.json             |  66 +++
 src/ipa/rpi/vc4/data/imx477.json             |  71 +++
 src/ipa/rpi/vc4/data/imx708.json             |  74 +++
 src/ipa/rpi/vc4/data/imx708_wide.json        |  64 +++
 src/ipa/rpi/vc4/data/ov5647.json             |  66 +++
 24 files changed, 2152 insertions(+), 420 deletions(-)
 create mode 100644 src/ipa/rpi/controller/rpi/awb_bayes.cpp
 create mode 100644 src/ipa/rpi/controller/rpi/awb_nn.cpp