C++ Gamma校正

void gamma_correction(cv::Mat& src, cv::Mat& dst, float K) {
	uchar LUT[256];
	src.copyTo(dst);
	for (int i = 0; i < 256; i++) {
		//float f = (i + 0.5f) / 255;
		float f = i / 255.0;
		f = pow(f, K);
		//LUT[i] = cv::saturate_cast<uchar>(f*255.0f-0.5f);
		LUT[i] = cv::saturate_cast<uchar>(f*255.0);
	}

	if (dst.channels() == 1) {
		cv::MatIterator_<uchar> it = dst.begin<uchar>();
		cv::MatIterator_<uchar> it_end = dst.end<uchar>();
		for (; it != it_end; ++it) {
			*it = LUT[(*it)];
		}
	}
	else {
		cv::MatIterator_<cv::Vec3b> it = dst.begin<cv::Vec3b>();
		cv::MatIterator_<cv::Vec3b> it_end = dst.end<cv::Vec3b>();
		for (; it != it_end; ++it) {
			(*it)[0] = LUT[(*it)[0]];
			(*it)[1] = LUT[(*it)[1]];
			(*it)[2] = LUT[(*it)[2]];
		}
	}

}
 

猜你喜欢

转载自blog.csdn.net/qq_41598072/article/details/109821151