From d628e8eab7371f136cc05ae090b904eede7f9c55 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Tue, 5 Nov 2019 16:57:03 +0300 Subject: [PATCH] Fixed darknet.py for Uncertainty (gaussian_yolo_layer) --- build/darknet/x64/darknet.py | 3 ++- darknet.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/build/darknet/x64/darknet.py b/build/darknet/x64/darknet.py index 10c9a456..5cfb26ba 100644 --- a/build/darknet/x64/darknet.py +++ b/build/darknet/x64/darknet.py @@ -59,7 +59,8 @@ class DETECTION(Structure): ("prob", POINTER(c_float)), ("mask", POINTER(c_float)), ("objectness", c_float), - ("sort_class", c_int)] + ("sort_class", c_int), + ("uc", POINTER(c_float))] class IMAGE(Structure): diff --git a/darknet.py b/darknet.py index 10c9a456..5cfb26ba 100644 --- a/darknet.py +++ b/darknet.py @@ -59,7 +59,8 @@ class DETECTION(Structure): ("prob", POINTER(c_float)), ("mask", POINTER(c_float)), ("objectness", c_float), - ("sort_class", c_int)] + ("sort_class", c_int), + ("uc", POINTER(c_float))] class IMAGE(Structure):