normalize and invert depth

src/main.rs

@@ -299,6 +299,21 @@ fn training() {
 		texture_data.clear();
 	}
 
+	// normalize inputs
+	let global_min = *inputs
+		.iter()
+		.min_by(|a, b| a.partial_cmp(b).unwrap())
+		.unwrap();
+	let global_max = *inputs
+		.iter()
+		.max_by(|a, b| a.partial_cmp(b).unwrap())
+		.unwrap();
+	let global_range = global_max - global_min;
+	println!("Normalizing to range {global_min} - {global_max}");
+	inputs.iter_mut().for_each(|value| {
+		*value = 1.0 - (*value - global_min) / global_range;
+	});
+
 	let device = burn::backend::cuda::CudaDevice::new(gpu_id);
 
 	let mut model: Net<TrainingBackend> = Net::init(&device);