diff --git a/darkflow/utils/loader.py b/darkflow/utils/loader.py index 723560df5..12ca69b26 100644 --- a/darkflow/utils/loader.py +++ b/darkflow/utils/loader.py @@ -114,11 +114,15 @@ def __init__(self, path): return else: self.size = os.path.getsize(path)# save the path - major, minor, revision, seen = np.memmap(path, + major, minor, revision = np.memmap(path, shape = (), mode = 'r', offset = 0, - dtype = '({})i4,'.format(4)) + dtype = '({})i4,'.format(3)) self.transpose = major > 1000 or minor > 1000 - self.offset = 16 + if ((major*10 + minor) >= 2 and major < 1000 and minor < 1000): + import ctypes + self.offset = 12 + ctypes.sizeof(ctypes.c_size_t) + else: + self.offset = 16 def walk(self, size): if self.eof: return None @@ -149,4 +153,4 @@ def model_name(file_path): num = int(file_name[-1]) return '-'.join(file_name[:-1]) if ext == 'weights': - return file_name \ No newline at end of file + return file_name