Commit 95647831 authored by Andrey Filippov's avatar Andrey Filippov

Attached ML generation and export of the DSI-aware warped images

parent 23452c5b
...@@ -1163,6 +1163,7 @@ public class ImageDtt extends ImageDttCPU { ...@@ -1163,6 +1163,7 @@ public class ImageDtt extends ImageDttCPU {
* Convert reference scene to FD and save result in extra GPU array for the future interscene correlation * Convert reference scene to FD and save result in extra GPU array for the future interscene correlation
* Geometry correction and images will come from gpuQuad instance - * Geometry correction and images will come from gpuQuad instance -
* @param imgdtt_params * @param imgdtt_params
* @param use_reference_buffer true - use extra GPU array, false - use main one
* @param tp_tasks * @param tp_tasks
* @param gpu_sigma_r * @param gpu_sigma_r
* @param gpu_sigma_b * @param gpu_sigma_b
...@@ -1173,6 +1174,7 @@ public class ImageDtt extends ImageDttCPU { ...@@ -1173,6 +1174,7 @@ public class ImageDtt extends ImageDttCPU {
*/ */
public void setReferenceTD( public void setReferenceTD(
final ImageDttParameters imgdtt_params, // Now just extra correlation parameters, later will include, most others final ImageDttParameters imgdtt_params, // Now just extra correlation parameters, later will include, most others
final boolean use_reference_buffer,
final TpTask[] tp_tasks, final TpTask[] tp_tasks,
final double gpu_sigma_r, // 0.9, 1.1 final double gpu_sigma_r, // 0.9, 1.1
final double gpu_sigma_b, // 0.9, 1.1 final double gpu_sigma_b, // 0.9, 1.1
...@@ -1201,7 +1203,7 @@ public class ImageDtt extends ImageDttCPU { ...@@ -1201,7 +1203,7 @@ public class ImageDtt extends ImageDttCPU {
tp_tasks, tp_tasks,
false); // boolean use_aux // while is it in class member? - just to be able to free false); // boolean use_aux // while is it in class member? - just to be able to free
// Skipping if ((fdisp_dist != null) || (fpxpy != null)) {... // Skipping if ((fdisp_dist != null) || (fpxpy != null)) {...
gpuQuad.execConvertDirect(true); // put results into a "reference" buffer gpuQuad.execConvertDirect(use_reference_buffer); // put results into a "reference" buffer
} }
......
...@@ -158,6 +158,18 @@ public class QuadCLTCPU { ...@@ -158,6 +158,18 @@ public class QuadCLTCPU {
public double [][] ds_from_main = null; public double [][] ds_from_main = null;
public double [][] dsrbg = null; // D, S, R,B,G public double [][] dsrbg = null; // D, S, R,B,G
public double [][] getDLS(){ // get disparity, disparity_lma, strength
if (dsi == null) {
// System.out.println("dsi== null, use spawnQuadCLT(), restoreFromModel(), ... to set it");
return null;
}
double [][] dls = new double[3][];
dls[0] = dsi[isAux()? TwoQuadCLT.DSI_DISPARITY_AUX : TwoQuadCLT.DSI_DISPARITY_MAIN];
dls[1] = dsi[isAux()? TwoQuadCLT.DSI_DISPARITY_AUX_LMA : TwoQuadCLT.DSI_DISPARITY_MAIN_LMA];
dls[2] = dsi[isAux()? TwoQuadCLT.DSI_STRENGTH_AUX : TwoQuadCLT.DSI_STRENGTH_MAIN];
return dls;
}
public TileProcessor getTileProcessor() { public TileProcessor getTileProcessor() {
return tp; return tp;
} }
...@@ -502,8 +514,6 @@ public class QuadCLTCPU { ...@@ -502,8 +514,6 @@ public class QuadCLTCPU {
} }
public String [] getDSRGGTitles() { public String [] getDSRGGTitles() {
return isMonochrome()? return isMonochrome()?
// (new String[]{"disparity","strength", "disparity_lma","Y"}):
// (new String[]{"disparity","strength", "disparity_lma","R","B","G"});
(new String[]{"disparity","strength", "Y"}): (new String[]{"disparity","strength", "Y"}):
(new String[]{"disparity","strength", "R","B","G"}); (new String[]{"disparity","strength", "R","B","G"});
} }
...@@ -643,7 +653,7 @@ public class QuadCLTCPU { ...@@ -643,7 +653,7 @@ public class QuadCLTCPU {
return rgba; return rgba;
} }
public QuadCLTCPU restoreFromModel( public QuadCLTCPU restoreFromModel( // restores dsi
CLTParameters clt_parameters, CLTParameters clt_parameters,
ColorProcParameters colorProcParameters, ColorProcParameters colorProcParameters,
NoiseParameters noise_sigma_level, NoiseParameters noise_sigma_level,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment