Skip to content

Commit

Permalink
rtl cecommended changes mae in AE and AWB
Browse files Browse the repository at this point in the history
  • Loading branch information
bakhtawar-10xe committed Sep 21, 2023
1 parent 9bdb4c0 commit 98c257c
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 13 deletions.
34 changes: 22 additions & 12 deletions modules/auto_exposure.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ def apply_window_offset_crop(self):
"""
Get AE Stats window by cropping the offsets
"""
offsets = np.ceil(self.stats_window_offset / 4) * 4
offsets = self.stats_window_offset
top = int(offsets[0])
bottom = None if offsets[1] == 0 else -int(offsets[1])
left = int(offsets[2])
Expand Down Expand Up @@ -104,7 +104,7 @@ def get_luminance_histogram_skewness(self, img):
Zwillinger, D. and Kokoska, S. (2000). CRC Standard Probability and Statistics
Tables and Formulae. Chapman & Hall: New York. 2000. Section 2.2.24.1
"""

img_orig = np.copy(img)
# First subtract central luminance to calculate skewness around it
img = img.astype(np.float64) - self.center_illuminance

Expand All @@ -120,18 +120,28 @@ def get_luminance_histogram_skewness(self, img):
if self.is_debug:
print(" - AE - Actual_Skewness = ", skewness)

sign_m3 = np.sign(m_3)

m_2 = m_2 >> 6
m_3 = abs(m_3) >> 9

approx_sqrt_m_2 = approx_sqrt(m_2)
new_skewness, _ = get_approximate(m_3 / (m_2 * approx_sqrt_m_2), 16, 8)
new_skewness = sign_m3 * new_skewness
# all integer calc
img_int = img_orig.astype(np.int64) - self.center_illuminance
img_int_size = img_int.size
m_2_int = np.sum(np.power(img_int, 2)).astype(np.int64)
m_3_int = np.sum(np.power(img_int, 3)).astype(np.int64)
m_2_int = np.int64(m_2_int / img_int_size)
m_3_int = np.int64(m_3_int / img_int_size)
sign_m3_int = np.sign(m_3_int)
# all integer calc

m_2_int = m_2_int >> 6
m_3_int = abs(m_3_int) >> 9

approx_sqrt_m_2_int = approx_sqrt(m_2_int)
new_skewness_int = (
np.int64((m_3_int * 256) / (m_2_int * approx_sqrt_m_2_int)) / 256
)
# new_skewness_int = sign_m3_int * new_skewness_int
if self.is_debug:
print(" - AE - Approx_Skewness = ", new_skewness)
print(" - AE - Approx_Skewness Int = ", new_skewness_int)

return new_skewness
return new_skewness_int

def execute(self):
"""
Expand Down
2 changes: 1 addition & 1 deletion modules/auto_white_balance.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def apply_window_offset_crop(self):
"""
Get AWB Stats window by cropping the offsets
"""
offsets = np.ceil(self.stats_window_offset / 4) * 4
offsets = self.stats_window_offset
top = int(offsets[0])
bottom = None if offsets[1] == 0 else -int(offsets[1])
left = int(offsets[2])
Expand Down

0 comments on commit 98c257c

Please sign in to comment.