Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[feat]IBA:demosaic add white balancing #4499

Open
wants to merge 10 commits into
base: main
Choose a base branch
from

Conversation

antond-weta
Copy link
Contributor

@antond-weta antond-weta commented Oct 22, 2024

Description

Adds white balancing functionality to IBA:demosaic

Tests

Updated unittests

Checklist:

  • I have read the contribution guidelines.
  • I have updated the documentation, if applicable. (Check if there is no
    need to update the documentation, for example if this is a bug fix that
    doesn't change the API.)
  • I have ensured that the change is tested somewhere in the testsuite
    (adding new test cases if necessary).
  • If I added or modified a C++ API call, I have also amended the
    corresponding Python bindings (and if altering ImageBufAlgo functions, also
    exposed the new functionality as oiiotool options).
  • My code follows the prevailing code style of this project. If I haven't
    already run clang-format before submitting, I definitely will look at the CI
    test that runs clang-format and fix anything that it highlights as being
    nonconforming.

@antond-weta antond-weta marked this pull request as draft October 22, 2024 02:11
Signed-off-by: Anton Dukhovnikov <[email protected]>
Signed-off-by: Anton Dukhovnikov <[email protected]>
Copy link
Collaborator

@lgritz lgritz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This LGTM. It's still marked as "draft". Is it ready to be reviewed and merged?

@antond-weta
Copy link
Contributor Author

Hi Larry, if you're happy with the code, I just need to update the docs and add some tests.

I've also just realised we could add an "auto" or "meta" value to the Bayer layout and white balance parameters, which would pull the values from the ImageBuf attributes.

@lgritz
Copy link
Collaborator

lgritz commented Oct 25, 2024

I'm happy with the code. Docs+tests should take it over the finish line.

I'm not quite sure what you mean about the ImageBuf attributes? Do you mean that when the image is read in, you expect the raw reader would drop hints in the spec about the default white balancing method to use, so IBA::demosaic should honor that if it gets no other overriding instructions? Or did you mean something different?

@antond-weta
Copy link
Contributor Author

Yep, exactly that. The raw reader currently adds the "raw:FilterPattern" attribute to the output when asked not to debayer. We could do something similar for white balancing. We can leave this for later, as adding an auto white balancing mode should not change the default behaviour, I think. If we were to add the "auto" layout pattern, it should become the default, as the current default behaviour is to assume RGGB. This is beyond the scope of this change though.

I'll update the docs, tests and the python bindings.

Signed-off-by: Anton Dukhovnikov <[email protected]>
@antond-weta
Copy link
Contributor Author

Done!
Larry, could you please check my python bindings, I'm not sure if there is a better way.

@antond-weta antond-weta marked this pull request as ready for review October 27, 2024 01:19
Copy link
Collaborator

@lgritz lgritz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also, I think before this is ready to merge, we will need the docs to be updated for all of C++ (imagebufalgo.h), Python (pythonbindings.rst), and oiiotool (oiiotool.rst), to explain the new parameters.

Comment on lines +1169 to +1173
float wb[4] = {2.0, 0.8, 1.2, 1.5};
ParamValue options[] = {
{ "layout", "GRBG" },
ParamValue("white-balance", TypeFloat, 4, wb)
};
Copy link
Collaborator

@lgritz lgritz Oct 27, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Not requesting this as part of this PR, but just noting that at some point, we should do the thing where we move these actual code snippets to the appropriate parts of testsuite/docs-examples-cpp and testsuite/docs-examples-python, and then just include them by reference in the docs, so that it is confirmed that they compile and work correct with every CI run.

Comment on lines 3471 to 3479
string_view str(wb);
float white_balance[4] = { 1.0f, 1.0f, 1.0f, 1.0f };
if (str.size() && Strutil::parse_float(str, white_balance[0])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[1])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[2])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[3])) {
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think there is a more compact alternative, and also more correct, because it will fail if the string contains any extra non-whitespace after the 4th value.

Suggested change
string_view str(wb);
float white_balance[4] = { 1.0f, 1.0f, 1.0f, 1.0f };
if (str.size() && Strutil::parse_float(str, white_balance[0])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[1])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[2])
&& Strutil::parse_char(str, ',')
&& Strutil::parse_float(str, white_balance[3])) {
string_view str(wb);
Strutil::remove_trailing_whitespace(str);
float white_balance[4] = { 1.0f, 1.0f, 1.0f, 1.0f };
if (Strutil::parse_values(str, "", white_balance, ",") && str.size()) {

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done! I've also decided to allow 3 values, for RGB, and extend them to RGGB automatically.

Comment on lines 3115 to 3116
"wb_r"_a = 1.0f, "wb_g1"_a = 1.0f, "wb_g2"_a = 1.0f,
"wb_b"_a = 1.0f, "roi"_a = ROI::All(), "nthreads"_a = 0)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This part, I'm not so sure about. Would it be better to pass white_balance as one parameter that expects a tuple or array (checked to be 4 floats), versus 4 separate float parameters?

I don't have a strong feeling one way or the other. I'm just wondering if that would be more symmetric with the C++ API, and/or more idiomatic for a Python native speaker?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I've changed that to use tuples, seems to be more inline with other python bindings

testsuite/oiiotool-demosaic/run.py Show resolved Hide resolved
Signed-off-by: Anton Dukhovnikov <[email protected]>
Signed-off-by: Anton Dukhovnikov <[email protected]>
Signed-off-by: Anton Dukhovnikov <[email protected]>
src/doc/pythonbindings.rst Outdated Show resolved Hide resolved
Comment on lines +2166 to +2168
/// - "white-balance" : float[3] or float[4] (default: {1.0, 1.0, 1.0, 1.0})
///
/// Optional white-balancing weights. Can contain either three (RGB), or four (RGGB) values.
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a use case for specifying two green values?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I have seen some images recently with the greens in red and blue rows having distinctly different levels. Sony A99, I think.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a use case for specifying two green values?

RAW_PENTAX_K200D.PEF in the test suite has 2 different weights for the green sub-channels

src/doc/oiiotool.rst Outdated Show resolved Hide resolved
@lgritz
Copy link
Collaborator

lgritz commented Oct 29, 2024

I'm confused about the white balance values.

  1. When you do give 4 values, are they always in the order RGGB, or are they in the order implied by the "layout"? I think we need to be more clear about this.

  2. Will anybody ever really want 4 values, with two different white balance values for the two greens? If not, a lot of things get easier if we just document the type as a Color3 instead of an array of floats that might be either 3 or 4 elements long.

antond-weta and others added 2 commits October 30, 2024 08:20
Co-authored-by: Larry Gritz <[email protected]>
Signed-off-by: Anton Dukhovnikov <[email protected]>
Co-authored-by: Larry Gritz <[email protected]>
Signed-off-by: Anton Dukhovnikov <[email protected]>
@antond-weta
Copy link
Contributor Author

Hi Larry,

I've been mostly following exiftool, which shows WB as either RGB or RGGB, regardless of the layout:

exiftool G85.RW2 | grep 'WB RG'
WB RGB Levels 1                 : 612 256 399
WB RGB Levels 2                 : 657 256 372
WB RGB Levels 3                 : 703 256 354
WB RGB Levels 4                 : 420 256 575
WB RGB Levels 5                 : 632 256 393
WB RGB Levels 6                 : 576 256 400
WB RGB Levels 7                 : 420 256 575
exiftool R5.CR3 | grep 'WB RG'
WB RGGB Levels As Shot          : 2241 1024 1024 1771
WB RGGB Levels Auto             : 2241 1024 1024 1771
WB RGGB Levels Measured         : 2241 1024 1024 1771
WB RGGB Levels Daylight         : 1960 1024 1024 1913
WB RGGB Levels Shade            : 2294 1024 1024 1636
WB RGGB Levels Cloudy           : 2118 1024 1024 1777
WB RGGB Levels Tungsten         : 1348 1024 1024 2962
WB RGGB Levels Fluorescent      : 1667 1024 1024 2819
WB RGGB Levels Kelvin           : 1960 1024 1024 1913
WB RGGB Levels Flash            : 2203 1024 1024 1708
WB RGGB Levels                  : 2241 1024 1024 1771

Libraw stores the weights as RGBG, regardless of the layout. I find the RGGB easier to understand, because the order also represents a valid Bayer layout, so it is clear where each G sits.

@lgritz
Copy link
Collaborator

lgritz commented Oct 29, 2024

Great, precedence from another widely used tool is a good reason for doing it a particular way.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants