diff --git a/scripts/libraries/ml/ml/postprocessing/darknet.py b/scripts/libraries/ml/ml/postprocessing/darknet.py index 159495e32..f23a05e86 100644 --- a/scripts/libraries/ml/ml/postprocessing/darknet.py +++ b/scripts/libraries/ml/ml/postprocessing/darknet.py @@ -31,6 +31,9 @@ from micropython import const from ulab import numpy as np +_NO_DETECTION = const(()) + + # This is a lightweight version of the tiny yolo v2 object detection algorithm. # It was optimized to work well on embedded devices with limited computational resources. class YoloV2: diff --git a/scripts/libraries/ml/ml/postprocessing/edgeimpulse.py b/scripts/libraries/ml/ml/postprocessing/edgeimpulse.py index 18c95f55e..d2d139b78 100644 --- a/scripts/libraries/ml/ml/postprocessing/edgeimpulse.py +++ b/scripts/libraries/ml/ml/postprocessing/edgeimpulse.py @@ -31,6 +31,9 @@ from micropython import const from ulab import numpy as np +_NO_DETECTION = const(()) + + class Fomo: _FOMO_CLASSES = const(1) diff --git a/scripts/libraries/ml/ml/postprocessing/mediapipe.py b/scripts/libraries/ml/ml/postprocessing/mediapipe.py index 600cc1bb4..b80635958 100644 --- a/scripts/libraries/ml/ml/postprocessing/mediapipe.py +++ b/scripts/libraries/ml/ml/postprocessing/mediapipe.py @@ -31,6 +31,9 @@ from micropython import const from ulab import numpy as np +_NO_DETECTION = const(()) + + class BlazeFace: _BLAZEFACE_CX = const(0) _BLAZEFACE_CY = const(1) diff --git a/scripts/libraries/ml/ml/postprocessing/ultralytics.py b/scripts/libraries/ml/ml/postprocessing/ultralytics.py index 88783dd54..f7e7a71bc 100644 --- a/scripts/libraries/ml/ml/postprocessing/ultralytics.py +++ b/scripts/libraries/ml/ml/postprocessing/ultralytics.py @@ -31,6 +31,9 @@ from micropython import const from ulab import numpy as np +_NO_DETECTION = const(()) + + class YoloV5: _YOLO_V5_CX = const(0) _YOLO_V5_CY = const(1) diff --git a/scripts/libraries/ml/ml/utils.py b/scripts/libraries/ml/ml/utils.py index d55283203..b481919b1 100644 --- a/scripts/libraries/ml/ml/utils.py +++ b/scripts/libraries/ml/ml/utils.py @@ -30,9 +30,6 @@ import math from ulab import numpy as np -_NO_DETECTION = const(()) - - def logit(x): return np.log(x / (1.0 - x))