Fixed darknet.py for Uncertainty (gaussian_yolo_layer)

6 years ago · d628e8eab7
parent e345b8793c
commit d628e8eab7
2 changed files with 4 additions and 2 deletions
--- 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):
--- 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):