Fix prescreener with float inputs

sekrit-twc · Sep 22, 2021 · bbdabfd · bbdabfd
1 parent 380c58e
commit bbdabfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/znedi3/znedi3_impl.cpp b/znedi3/znedi3_impl.cpp
@@ -125,7 +125,7 @@ znedi3_filter::znedi3_filter(const NNEDI3Weights &weights, const znedi3_filter_p
 	assert(model_set.find(traits) != model_set.end());
 	const PredictorModel &model = *model_set.find(traits);
 
-	double pixel_half_val = static_cast<double>((1UL << bit_depth) - 1) / 2.0;
+	double pixel_half_val = (m_type == PixelType::HALF || m_type == PixelType::FLOAT) ? 0.5 : static_cast<double>((1UL << bit_depth) - 1) / 2.0;
 
 	switch (params.prescreen) {
 	case ZNEDI3_PRESCREEN_NONE: