scripts/libraries: Fix issue with using YOLOV2.

This commit is contained in:
Kwabena W. Agyeman 2025-10-19 15:57:58 +04:00
parent 7cbdb927da
commit c4b0b5a3dc

View File

@ -47,7 +47,6 @@ class YoloV2:
def __init__(self, threshold=0.6, anchors=None, nms_threshold=0.1, nms_sigma=0.1):
self.threshold = threshold
self.anchors = anchors
self.anchors_len = len(self.anchors)
self.nms_threshold = nms_threshold
self.nms_sigma = nms_sigma
@ -58,6 +57,8 @@ class YoloV2:
[5.55170, 9.30660],
[9.72600, 11.1422]])
self.anchors_len = len(self.anchors)
def __call__(self, model, inputs, outputs):
def softmax(x):