Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

optimize imports

This commit is contained in:
Glenn Jocher 2019-02-15 14:07:55 +02:00
parent 9b4e9924fb
commit ee4abc8cdf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -316,7 +316,7 @@ def load_darknet_weights(self, weights, cutoff=-1):
# Needed to write header when saving weights # Needed to write header when saving weights
self.header_info = header self.header_info = header
self.seen = header[3] self.seen = header[3] # number of images seen during training
weights = np.fromfile(fp, dtype=np.float32) # The rest are weights weights = np.fromfile(fp, dtype=np.float32) # The rest are weights
fp.close() fp.close()
@ -365,7 +365,7 @@ def load_darknet_weights(self, weights, cutoff=-1):
def save_weights(self, path, cutoff=-1): def save_weights(self, path, cutoff=-1):
fp = open(path, 'wb') fp = open(path, 'wb')
self.header_info[3] = self.seen self.header_info[3] = self.seen # number of images seen during training
self.header_info.tofile(fp) self.header_info.tofile(fp)
# Iterate through layers # Iterate through layers