perform this conversion from tensors

[github-actions]

burn_dino/crates/bevy_burn_dino/src/main.rs

Line 348 in 0a59eb7

// TODO: perform this conversion from tensors

}


#[cfg(feature = "web")]
mod web {
    use std::cell::RefCell;

    use image::{
        DynamicImage,
        RgbImage,
        RgbaImage,
    };
    use wasm_bindgen::prelude::*;

    thread_local! {
        pub static SAMPLE_RECEIVER: RefCell<Option<RgbImage>> = RefCell::new(None);
    }

    #[wasm_bindgen]
    pub fn frame_input(pixel_data: &[u8], width: u32, height: u32) {
        let rgba_image = RgbaImage::from_raw(width, height, pixel_data.to_vec())
            .expect("failed to create RgbImage");

        // TODO: perform video element -> burn's webgpu texture conversion directly
        // TODO: perform this conversion from tensors
        let dynamic_image = DynamicImage::ImageRgba8(rgba_image);
        let rgb_image: RgbImage = dynamic_image.to_rgb8();

        SAMPLE_RECEIVER.with(|receiver| {
            *receiver.borrow_mut() = Some(rgb_image);
        });
    }
}


#[derive(Resource)]
struct DinoModel<B: Backend> {
    config: DinoVisionTransformerConfig,
    device: B::Device,
    model: Arc<Mutex<DinoVisionTransformer<B>>>,
}

#[derive(Resource)]