mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
scripts/libraries: Fix issue with using YOLOV2.
This commit is contained in:
parent
7cbdb927da
commit
c4b0b5a3dc
@ -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):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user