Skip to content

Instantly share code, notes, and snippets.

@igor-panteleev
Created December 7, 2017 10:59
Show Gist options
  • Save igor-panteleev/00f9dd3403a2d5b1a8afeb064c6d74d7 to your computer and use it in GitHub Desktop.
Save igor-panteleev/00f9dd3403a2d5b1a8afeb064c6d74d7 to your computer and use it in GitHub Desktop.
devicehive-video-analysis DarkFlow modifications
# darkflow/net/build.py
def saveckpt(self):
self.saver.save(self.sess, './{}_model.ckpt'.format(self.meta['name']))
# darkflow/cli.py
if FLAGS.saveckpt:
print('Rebuild a constant version ...')
tfnet.saveckpt(); exit('Done')
# darkflow/cli.py
self.define('saveckpt', False, 'save net and weight to a .ckpt file')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment