Skip to content

Commit a91e3a5

Browse files
committed
add new field boundary model
1 parent a92f38f commit a91e3a5

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

deploy/models/field_boundary.onnx

5.43 KB
Binary file not shown.

yggdrasil/src/vision/field_boundary.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -195,8 +195,6 @@ pub fn detect_field_boundary(
195195
.resize(MODEL_INPUT_WIDTH, MODEL_INPUT_HEIGHT)
196196
.expect("Failed to resize image")
197197
.into_iter()
198-
// TODO: Retrain the model in u8 inputs
199-
.map(f32::from)
200198
.collect::<Vec<_>>();
201199

202200
commands
@@ -224,7 +222,7 @@ pub fn detect_field_boundary(
224222
pub struct FieldBoundaryModel;
225223

226224
impl MlModel for FieldBoundaryModel {
227-
type Inputs = Vec<f32>;
225+
type Inputs = Vec<u8>;
228226
type Outputs = Vec<f32>;
229227

230228
const ONNX_PATH: &'static str = "models/field_boundary.onnx";

0 commit comments

Comments
 (0)