AutoWhiteBalance icon indicating copy to clipboard operation
AutoWhiteBalance copied to clipboard

Is there any limitation about FFCC algorithms?

Open gonglixue opened this issue 2 years ago • 0 comments

Hi yuan, thanks for your sharing.

I used your code to test my own images while the results look not good. Is there any limitation about this method? PNG16 raw images are used for training. Does it work for normal 8bit image?

  • input image:
image
  • output image: image

(I modified the funcion name in main.cpp. Is it a correct usage? )

int main(int argc, char* argv[]) {
	AutoWhiteBalance autoWB;
	autoWB.loadModel("E:/Project/AutoWhiteBalance/data/model/model.bin");

	cv::Mat img = cv::imread("E:/data/giga/NanshanIPark/2/calibrate/ref_00.jpg");
	cv::cuda::GpuMat img_d;
	img_d.upload(img);

	float gain_r, gain_g, gain_b;

	time_t begin, end;
	begin = clock();

	// autoWB.apply(img_d, gain_r, gain_g, gain_b);
	// autoWB.apply(img_d, gain_r, gain_g, gain_b);

        autoWB.calc(img_d, gain_r, gain_g, gain_b);

	end = clock();
	printf("Auto white balance update, cost %f milliseconds ...\n",
		static_cast<float>(end - begin) / static_cast<double>(CLOCKS_PER_SEC) * 1000);

	autoWB.applyWhiteBalance(img_d, gain_r, gain_g, gain_b);

	cv::Mat img2;
	img_d.download(img2);

	return 0;
}

gonglixue avatar Oct 11 '22 09:10 gonglixue