CameronJGrant/strafe-ai
1
Fork 0
forked from StrafesNET/strafe-ai
Code Pull Requests Activity

png

Browse Source
This commit is contained in:
Rhys Lloyd
2026-03-26 13:29:59 -07:00
parent 1f7dbbfee9
commit 5902f4fb1b
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.rs
View File

@@ -15,12 +15,9 @@ pub fn output_image_native(image_data: &[u8], texture_dims: (usize, usize), path
use std::io::Write;
let mut png_data = Vec::<u8>::with_capacity(image_data.len());
let mut encoder = png::Encoder::new(
std::io::Cursor::new(&mut png_data),
texture_dims.0 as u32,
texture_dims.1 as u32,
);
encoder.set_color(png::ColorType::Rgba);
let mut encoder =
png::Encoder::new(&mut png_data, texture_dims.0 as u32, texture_dims.1 as u32);
encoder.set_color(png::ColorType::Grayscale);
let mut png_writer = encoder.write_header().unwrap();
png_writer.write_image_data(image_data).unwrap();
png_writer.finish().unwrap();
@@ -324,9 +321,14 @@ fn training() {
)
}
let input_slice = &inputs[i * INPUT..(i + 1) * INPUT];
let u8_slice = unsafe {
core::slice::from_raw_parts(input_slice.as_ptr().cast(), INPUT * size_of::<f32>())
};
// write a png
output_image_native(
&inputs[i * input_size..(i + 1) * input_size],
u8_slice,
(SIZE_X, SIZE_Y),
format!("depth_images/{i}.png").into(),
);