diff --git a/build/darknet/YoloWrapper.cs b/build/darknet/YoloWrapper.cs index ade00597..14bf56d4 100644 --- a/build/darknet/YoloWrapper.cs +++ b/build/darknet/YoloWrapper.cs @@ -6,6 +6,7 @@ namespace Darknet public class YoloWrapper : IDisposable { private const string YoloLibraryName = "yolo_cpp_dll.dll"; + private const int MaxObjects = 1000; [DllImport(YoloLibraryName, EntryPoint = "init")] public static extern int InitializeYolo(string configurationFilename, string weightsFilename, int gpu); @@ -29,10 +30,10 @@ namespace Darknet public UInt32 frames_counter; }; - [StructLayout(LayoutKind.Sequential, Size = 10)] + [StructLayout(LayoutKind.Sequential)] public struct BboxContainer { - [MarshalAs(UnmanagedType.ByValArray, SizeConst = 10)] + [MarshalAs(UnmanagedType.ByValArray, SizeConst = MaxObjects)] public bbox_t[] candidates; } diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp index c518fd77..4df9be5d 100644 --- a/src/yolo_v2_class.cpp +++ b/src/yolo_v2_class.cpp @@ -22,8 +22,6 @@ extern "C" { #define FRAMES 3 -int max_objects() { return C_SHARP_MAX_OBJECTS; } - //static Detector* detector = NULL; static std::unique_ptr detector; diff --git a/src/yolo_v2_class.hpp b/src/yolo_v2_class.hpp index 98758783..8f7f6cab 100644 --- a/src/yolo_v2_class.hpp +++ b/src/yolo_v2_class.hpp @@ -45,7 +45,6 @@ struct bbox_t_container { #include "opencv2/imgproc/imgproc_c.h" // C #endif // OPENCV -extern "C" YOLODLL_API int max_objects(); extern "C" YOLODLL_API int init(const char *configurationFilename, const char *weightsFilename, int gpu); extern "C" YOLODLL_API int detect_image(const char *filename, bbox_t_container &container); extern "C" YOLODLL_API int detect_mat(const uint8_t* data, const size_t data_length, bbox_t_container &container);