diff --git a/src/main.rs b/src/main.rs
index 2043e36..7d03b3b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -328,7 +328,7 @@ fn training() {
 			&inputs[i * INPUT..(i + 1) * INPUT]
 				.iter()
 				.copied()
-				.map(|f| (f * 255.0) as u8)
+				.map(|f| f as u8)
 				.collect::<Vec<u8>>(),
 			(SIZE_X, SIZE_Y),
 			format!("depth_images/{i}.png").into(),