Support batch_size when loading the neural network

Author: sodre

src/darknet/py/network.pyx

@@ -32,9 +32,11 @@ cdef class Network:
                 return Network(config.name, weights.name)
 
 
-    def __cinit__(self, config_file, weights_file):
-        clear = 1
-        self._c_network = dn.load_network(config_file.encode(), weights_file.encode(), clear)
+    def __cinit__(self, config_file, weights_file, clear = True, batch_size = 1):
+        self._c_network = dn.load_network_custom(config_file.encode(),
+                                                 weights_file.encode(),
+                                                 1 if clear else 0,
+                                                 batch_size)
         if self._c_network is NULL:
             raise RuntimeError("Failed to create the DarkNet Network...")
 

src/libdarknet/__init__.pxd

@@ -59,6 +59,7 @@ cdef extern from "darknet.h":
         pass
 
     network* load_network(char* cfg_filename, char* weights_filename, int clear)
+    network* load_network_custom(char* cfg_filename, char* weights_filename, int clear, int batch_size)
     void free_network(network self)
 
     int network_width(network *self);