Fixed darknet.py for Uncertainty (gaussian_yolo_layer)

This commit is contained in:
AlexeyAB 2019-11-05 16:57:03 +03:00
parent e345b8793c
commit d628e8eab7
2 changed files with 4 additions and 2 deletions

View File

@ -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):

View File

@ -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):