tici camerart III: prehistoric visuals (#20012)
* reworked * add ref * f16 * faster * extended tm * fix hdr * Revert "extended tm" This reverts commit 55ba3ae3205f7e962bf2f36634b23a595700c9a8. * reduce static/temporal noise * split kernels to sync max * no print * cleanup qcom2 * optimize * tune DC * doesnt work * minor fix * adaptive * smoothen HCG switch * t * histogram ceiling * recalulate params * group_hold * less contrast * recalibrate ccm * better ae * better ae revised * made some changes * clean up * midtones * some improvements * more * cleanup * remove more junk * lgtm Co-authored-by: Comma Device <device@comma.ai>pull/59/head
ZwX1616
5 years ago
committed by
GitHub
parent
c747654f05
commit
d8ddc1208e
5 changed files with 220 additions and 754 deletions
@ -1,135 +1,216 @@ |
||||
const __constant float3 color_correction[3] = { |
||||
#pragma OPENCL EXTENSION cl_khr_fp16 : enable |
||||
|
||||
const half black_level = 42.0; |
||||
|
||||
const __constant half3 color_correction[3] = { |
||||
// post wb CCM |
||||
(float3)(1.44602146, -0.24727126, -0.0403062), |
||||
(float3)(-0.37658179, 1.26329038, -0.45978396), |
||||
(float3)(-0.06943967, -0.01601912, 1.50009016), |
||||
(half3)(1.25985206, -0.378923, -0.21356857), |
||||
(half3)(-0.11117607, 1.3962182, -0.46342976), |
||||
(half3)(-0.21523926, -0.13449348, 1.47665819), |
||||
}; |
||||
|
||||
const __constant half3 base_ccm[3] = { |
||||
(half3)(1,0,0), |
||||
(half3)(0,1,0), |
||||
(half3)(0,0,1), |
||||
}; |
||||
|
||||
float3 color_correct(float r, float g, float b) { |
||||
float3 ret = (0,0,0); |
||||
|
||||
ret += r * color_correction[0]; |
||||
ret += g * color_correction[1]; |
||||
ret += b * color_correction[2]; |
||||
ret = max(0.0, min(1.0, ret)); |
||||
// tone mapping params |
||||
const half cpk = 0.75; |
||||
const half cpb = 0.125; |
||||
const half cpxk = 0.01; |
||||
const half cpxb = 0.01; |
||||
|
||||
half mf(half x, half cp) { |
||||
half rk = 8.6 - 66*cp; |
||||
if (x > cp) { |
||||
return (rk * (x-cp) * (1-(cpk*cp+cpb)) * (1+1/(rk*(1-cp))) / (1+rk*(x-cp))) + cpk*cp + cpb; |
||||
} else if (x < cp) { |
||||
return (rk * (x-cp) * (cpk*cp+cpb) * (1+1/(rk*cp)) / (1-rk*(x-cp))) + cpk*cp + cpb; |
||||
} else { |
||||
return x; |
||||
} |
||||
} |
||||
|
||||
half3 color_correct(half3 rgb, int ggain) { |
||||
half3 ret = (0,0,0); |
||||
half cpx = clamp(0.03h, 0.1h, cpxb + cpxk * min(10, ggain)); |
||||
rgb.x = mf(rgb.x, cpx); |
||||
rgb.y = mf(rgb.y, cpx); |
||||
rgb.z = mf(rgb.z, cpx); |
||||
ret += (half)rgb.x * (color_correction[0]); |
||||
ret += (half)rgb.y * (color_correction[1]); |
||||
ret += (half)rgb.z * (color_correction[2]); |
||||
ret = clamp(0.0h, 255.0h, ret*255.0h); |
||||
return ret; |
||||
} |
||||
|
||||
uint int_from_10(const uchar * source, uint start, uint offset) { |
||||
// source: source |
||||
// start: starting address of 0 |
||||
// offset: 0 - 3 |
||||
half val_from_10(const uchar * source, int gx, int gy) { |
||||
// parse 10bit |
||||
int start = gy * FRAME_STRIDE + (5 * (gx / 4)); |
||||
int offset = gx % 4; |
||||
uint major = (uint)source[start + offset] << 2; |
||||
uint minor = (source[start + 4] >> (2 * offset)) & 3; |
||||
return major + minor; |
||||
} |
||||
half pv = (half)(major + minor); |
||||
|
||||
float to_normal(uint x, int gx, int gy) { |
||||
float pv = (float)(x); |
||||
const float black_level = 42.0; |
||||
pv = max(0.0, pv - black_level); |
||||
pv /= (1024.0f - black_level); |
||||
// normalize |
||||
pv = max(0.0h, pv - black_level); |
||||
pv *= 0.00101833h; // /= (1024.0f - black_level); |
||||
|
||||
// correct vignetting |
||||
if (CAM_NUM == 1) { // fcamera |
||||
gx = (gx - RGB_WIDTH/2); |
||||
gy = (gy - RGB_HEIGHT/2); |
||||
float r = pow(gx*gx + gy*gy, 0.825); |
||||
float s = 1 / (1-0.00000733*r); |
||||
float r = gx*gx + gy*gy; |
||||
half s; |
||||
if (r < 62500) { |
||||
s = (half)(1.0f + 0.0000008f*r); |
||||
} else if (r < 490000) { |
||||
s = (half)(0.9625f + 0.0000014f*r); |
||||
} else if (r < 1102500) { |
||||
s = (half)(1.26434f + 0.0000000000016f*r*r); |
||||
} else { |
||||
s = (half)(0.53503625f + 0.0000000000022f*r*r); |
||||
} |
||||
pv = s * pv; |
||||
} |
||||
pv = 20*pv / (1.0f + 20*pv); // reinhard |
||||
|
||||
pv = clamp(0.0h, 1.0h, pv); |
||||
return pv; |
||||
} |
||||
|
||||
half fabs_diff(half x, half y) { |
||||
return fabs(x-y); |
||||
} |
||||
|
||||
half phi(half x) { |
||||
// detection funtion |
||||
return 2 - x; |
||||
// if (x > 1) { |
||||
// return 1 / x; |
||||
// } else { |
||||
// return 2 - x; |
||||
// } |
||||
} |
||||
|
||||
__kernel void debayer10(const __global uchar * in, |
||||
__global uchar * out, |
||||
__local float * cached |
||||
__local half * cached, |
||||
uint ggain |
||||
) |
||||
{ |
||||
const int x_global = get_global_id(0); |
||||
const int y_global = get_global_id(1); |
||||
|
||||
// const int globalOffset = ; |
||||
|
||||
const int localRowLen = 2 + get_local_size(0); // 2 padding |
||||
const int x_local = get_local_id(0); |
||||
const int y_local = get_local_id(1); |
||||
const int x_local = get_local_id(0); // 0-15 |
||||
const int y_local = get_local_id(1); // 0-15 |
||||
const int localOffset = (y_local + 1) * localRowLen + x_local + 1; // max 18x18-1 |
||||
|
||||
const int localOffset = (y_local + 1) * localRowLen + x_local + 1; |
||||
int out_idx = 3 * x_global + 3 * y_global * RGB_WIDTH; |
||||
|
||||
// cache local pixels first |
||||
// saves memory access and avoids repeated normalization |
||||
uint globalStart_10 = y_global * FRAME_STRIDE + (5 * (x_global / 4)); |
||||
uint offset_10 = x_global % 4; |
||||
uint raw_val = int_from_10(in, globalStart_10, offset_10); |
||||
cached[localOffset] = to_normal(raw_val, x_global, y_global); |
||||
half pv = val_from_10(in, x_global, y_global); |
||||
cached[localOffset] = pv; |
||||
|
||||
// edges |
||||
if (x_global < 1 || x_global > RGB_WIDTH - 2 || y_global < 1 || y_global > RGB_HEIGHT - 2) { |
||||
barrier(CLK_LOCAL_MEM_FENCE); |
||||
// don't care |
||||
if (x_global < 1 || x_global >= RGB_WIDTH - 1 || y_global < 1 || y_global >= RGB_HEIGHT - 1) { |
||||
return; |
||||
} else { |
||||
int localColOffset = -1; |
||||
int globalColOffset = -1; |
||||
|
||||
// cache padding |
||||
if (x_local < 1) { |
||||
localColOffset = x_local; |
||||
globalColOffset = -1; |
||||
cached[(y_local + 1) * localRowLen + x_local] = to_normal(int_from_10(in, y_global * FRAME_STRIDE + (5 * ((x_global-1) / 4)), (offset_10 + 3) % 4), x_global, y_global); |
||||
} else if (x_local >= get_local_size(0) - 1) { |
||||
localColOffset = x_local + 2; |
||||
globalColOffset = 1; |
||||
cached[localOffset + 1] = to_normal(int_from_10(in, y_global * FRAME_STRIDE + (5 * ((x_global+1) / 4)), (offset_10 + 1) % 4), x_global, y_global); |
||||
} |
||||
} |
||||
|
||||
// cache padding |
||||
int localColOffset = -1; |
||||
int globalColOffset = -1; |
||||
|
||||
// cache padding |
||||
if (x_local < 1) { |
||||
localColOffset = x_local; |
||||
globalColOffset = -1; |
||||
cached[(y_local + 1) * localRowLen + x_local] = val_from_10(in, x_global-1, y_global); |
||||
} else if (x_local >= get_local_size(0) - 1) { |
||||
localColOffset = x_local + 2; |
||||
globalColOffset = 1; |
||||
cached[localOffset + 1] = val_from_10(in, x_global+1, y_global); |
||||
} |
||||
|
||||
if (y_local < 1) { |
||||
cached[y_local * localRowLen + x_local + 1] = to_normal(int_from_10(in, globalStart_10 - FRAME_STRIDE, offset_10), x_global, y_global); |
||||
if (localColOffset != -1) { |
||||
cached[y_local * localRowLen + localColOffset] = to_normal(int_from_10(in, (y_global-1) * FRAME_STRIDE + (5 * ((x_global+globalColOffset) / 4)), (offset_10+4+globalColOffset) % 4), x_global, y_global); |
||||
} |
||||
} else if (y_local >= get_local_size(1) - 1) { |
||||
cached[(y_local + 2) * localRowLen + x_local + 1] = to_normal(int_from_10(in, globalStart_10 + FRAME_STRIDE, offset_10), x_global, y_global); |
||||
if (localColOffset != -1) { |
||||
cached[(y_local + 2) * localRowLen + localColOffset] = to_normal(int_from_10(in, (y_global+1) * FRAME_STRIDE + (5 * ((x_global+globalColOffset) / 4)), (offset_10+4+globalColOffset) % 4), x_global, y_global); |
||||
} |
||||
if (y_local < 1) { |
||||
cached[y_local * localRowLen + x_local + 1] = val_from_10(in, x_global, y_global-1); |
||||
if (localColOffset != -1) { |
||||
cached[y_local * localRowLen + localColOffset] = val_from_10(in, x_global+globalColOffset, y_global-1); |
||||
} |
||||
} else if (y_local >= get_local_size(1) - 1) { |
||||
cached[(y_local + 2) * localRowLen + x_local + 1] = val_from_10(in, x_global, y_global+1); |
||||
if (localColOffset != -1) { |
||||
cached[(y_local + 2) * localRowLen + localColOffset] = val_from_10(in, x_global+globalColOffset, y_global+1); |
||||
} |
||||
} |
||||
|
||||
// sync |
||||
barrier(CLK_LOCAL_MEM_FENCE); |
||||
|
||||
// perform debayer |
||||
float r; |
||||
float g; |
||||
float b; |
||||
|
||||
if (x_global % 2 == 0) { |
||||
if (y_global % 2 == 0) { // G1 |
||||
r = (cached[localOffset - 1] + cached[localOffset + 1]) / 2.0f; |
||||
g = (cached[localOffset] + cached[localOffset + localRowLen + 1]) / 2.0f; |
||||
b = (cached[localOffset - localRowLen] + cached[localOffset + localRowLen]) / 2.0f; |
||||
} else { // B |
||||
r = (cached[localOffset - localRowLen - 1] + cached[localOffset - localRowLen + 1] + cached[localOffset + localRowLen - 1] + cached[localOffset + localRowLen + 1]) / 4.0f; |
||||
g = (cached[localOffset - localRowLen] + cached[localOffset + localRowLen] + cached[localOffset - 1] + cached[localOffset + 1]) / 4.0f; |
||||
b = cached[localOffset]; |
||||
} |
||||
// sync |
||||
barrier(CLK_LOCAL_MEM_FENCE); |
||||
|
||||
half d1 = cached[localOffset - localRowLen - 1]; |
||||
half d2 = cached[localOffset - localRowLen + 1]; |
||||
half d3 = cached[localOffset + localRowLen - 1]; |
||||
half d4 = cached[localOffset + localRowLen + 1]; |
||||
half n1 = cached[localOffset - localRowLen]; |
||||
half n2 = cached[localOffset + 1]; |
||||
half n3 = cached[localOffset + localRowLen]; |
||||
half n4 = cached[localOffset - 1]; |
||||
|
||||
half3 rgb; |
||||
|
||||
// a simplified version of https://opensignalprocessingjournal.com/contents/volumes/V6/TOSIGPJ-6-1/TOSIGPJ-6-1.pdf |
||||
if (x_global % 2 == 0) { |
||||
if (y_global % 2 == 0) { |
||||
rgb.y = pv; // G1(R) |
||||
half k1 = phi(fabs_diff(d1, pv) + fabs_diff(d2, pv)); |
||||
half k2 = phi(fabs_diff(d2, pv) + fabs_diff(d4, pv)); |
||||
half k3 = phi(fabs_diff(d3, pv) + fabs_diff(d4, pv)); |
||||
half k4 = phi(fabs_diff(d1, pv) + fabs_diff(d3, pv)); |
||||
// R_G1 |
||||
rgb.x = (k2*n2+k4*n4)/(k2+k4); |
||||
// B_G1 |
||||
rgb.z = (k1*n1+k3*n3)/(k1+k3); |
||||
} else { |
||||
rgb.z = pv; // B |
||||
half k1 = phi(fabs_diff(d1, d3) + fabs_diff(d2, d4)); |
||||
half k2 = phi(fabs_diff(n1, n4) + fabs_diff(n2, n3)); |
||||
half k3 = phi(fabs_diff(d1, d2) + fabs_diff(d3, d4)); |
||||
half k4 = phi(fabs_diff(n1, n2) + fabs_diff(n3, n4)); |
||||
// G_B |
||||
rgb.y = (k1*(n1+n3)*0.5+k3*(n2+n4)*0.5)/(k1+k3); |
||||
// R_B |
||||
rgb.x = (k2*(d2+d3)*0.5+k4*(d1+d4)*0.5)/(k2+k4); |
||||
} |
||||
} else { |
||||
if (y_global % 2 == 0) { |
||||
rgb.x = pv; // R |
||||
half k1 = phi(fabs_diff(d1, d3) + fabs_diff(d2, d4)); |
||||
half k2 = phi(fabs_diff(n1, n4) + fabs_diff(n2, n3)); |
||||
half k3 = phi(fabs_diff(d1, d2) + fabs_diff(d3, d4)); |
||||
half k4 = phi(fabs_diff(n1, n2) + fabs_diff(n3, n4)); |
||||
// G_R |
||||
rgb.y = (k1*(n1+n3)*0.5+k3*(n2+n4)*0.5)/(k1+k3); |
||||
// B_R |
||||
rgb.z = (k2*(d2+d3)*0.5+k4*(d1+d4)*0.5)/(k2+k4); |
||||
} else { |
||||
if (y_global % 2 == 0) { // R |
||||
r = cached[localOffset]; |
||||
g = (cached[localOffset - localRowLen] + cached[localOffset + localRowLen] + cached[localOffset - 1] + cached[localOffset + 1]) / 4.0f; |
||||
b = (cached[localOffset - localRowLen - 1] + cached[localOffset - localRowLen + 1] + cached[localOffset + localRowLen - 1] + cached[localOffset + localRowLen + 1]) / 4.0f; |
||||
} else { // G2 |
||||
r = (cached[localOffset - localRowLen] + cached[localOffset + localRowLen]) / 2.0f; |
||||
g = (cached[localOffset] + cached[localOffset - localRowLen - 1]) / 2.0f; |
||||
b = (cached[localOffset - 1] + cached[localOffset + 1]) / 2.0f; |
||||
} |
||||
rgb.y = pv; // G2(B) |
||||
half k1 = phi(fabs_diff(d1, pv) + fabs_diff(d2, pv)); |
||||
half k2 = phi(fabs_diff(d2, pv) + fabs_diff(d4, pv)); |
||||
half k3 = phi(fabs_diff(d3, pv) + fabs_diff(d4, pv)); |
||||
half k4 = phi(fabs_diff(d1, pv) + fabs_diff(d3, pv)); |
||||
// R_G2 |
||||
rgb.x = (k1*n1+k3*n3)/(k1+k3); |
||||
// B_G2 |
||||
rgb.z = (k2*n2+k4*n4)/(k2+k4); |
||||
} |
||||
} |
||||
|
||||
float3 rgb = color_correct(r, g, b); |
||||
// rgb = srgb_gamma(rgb); |
||||
rgb = clamp(0.0h, 1.0h, rgb); |
||||
rgb = color_correct(rgb, (int)ggain); |
||||
|
||||
out[out_idx + 0] = (uchar)(rgb.z); |
||||
out[out_idx + 1] = (uchar)(rgb.y); |
||||
out[out_idx + 2] = (uchar)(rgb.x); |
||||
|
||||
// BGR output |
||||
out[3 * x_global + 3 * y_global * RGB_WIDTH + 0] = (uchar)(255.0f * rgb.z); |
||||
out[3 * x_global + 3 * y_global * RGB_WIDTH + 1] = (uchar)(255.0f * rgb.y); |
||||
out[3 * x_global + 3 * y_global * RGB_WIDTH + 2] = (uchar)(255.0f * rgb.x); |
||||
} |
||||
} |
||||
|
||||
Loading…
Reference in new issue