diff --git a/deploy/models/field_boundary.onnx b/deploy/models/field_boundary.onnx
index d3f03f881..ae7a1044d 100644
Binary files a/deploy/models/field_boundary.onnx and b/deploy/models/field_boundary.onnx differ
diff --git a/yggdrasil/src/vision/field_boundary.rs b/yggdrasil/src/vision/field_boundary.rs
index 826f76c21..327d1a52f 100644
--- a/yggdrasil/src/vision/field_boundary.rs
+++ b/yggdrasil/src/vision/field_boundary.rs
@@ -195,8 +195,6 @@ pub fn detect_field_boundary(
         .resize(MODEL_INPUT_WIDTH, MODEL_INPUT_HEIGHT)
         .expect("Failed to resize image")
         .into_iter()
-        // TODO: Retrain the model in u8 inputs
-        .map(f32::from)
         .collect::<Vec<_>>();
 
     commands
@@ -224,7 +222,7 @@ pub fn detect_field_boundary(
 pub struct FieldBoundaryModel;
 
 impl MlModel for FieldBoundaryModel {
-    type Inputs = Vec<f32>;
+    type Inputs = Vec<u8>;
     type Outputs = Vec<f32>;
 
     const ONNX_PATH: &'static str = "models/field_boundary.onnx";