Skip to content

Commit 2ba1b5b

Browse files
authored
Support object-detection model type with hyphen (#19)
1 parent aec4c37 commit 2ba1b5b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/video/imp.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -565,7 +565,8 @@ impl BaseTransformImpl for EdgeImpulseVideoInfer {
565565
let _model_height = params.image_input_height;
566566
let channels = params.image_channel_count;
567567
let is_object_detection = params.model_type == "constrained_object_detection"
568-
|| params.model_type == "object_detection";
568+
|| params.model_type == "object_detection"
569+
|| params.model_type == "object-detection";
569570
let is_anomaly_detection = params.model_type == "anomaly_detection"
570571
|| matches!(
571572
params.has_anomaly,

0 commit comments

Comments
 (0)