We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a92f38f commit a91e3a5Copy full SHA for a91e3a5
2 files changed
deploy/models/field_boundary.onnx
5.43 KB
yggdrasil/src/vision/field_boundary.rs
@@ -195,8 +195,6 @@ pub fn detect_field_boundary(
195
.resize(MODEL_INPUT_WIDTH, MODEL_INPUT_HEIGHT)
196
.expect("Failed to resize image")
197
.into_iter()
198
- // TODO: Retrain the model in u8 inputs
199
- .map(f32::from)
200
.collect::<Vec<_>>();
201
202
commands
@@ -224,7 +222,7 @@ pub fn detect_field_boundary(
224
222
pub struct FieldBoundaryModel;
225
223
226
impl MlModel for FieldBoundaryModel {
227
- type Inputs = Vec<f32>;
+ type Inputs = Vec<u8>;
228
type Outputs = Vec<f32>;
229
230
const ONNX_PATH: &'static str = "models/field_boundary.onnx";
0 commit comments