Skip to content

Instantly share code, notes, and snippets.

@igor-panteleev

igor-panteleev/build.py

Created December 7, 2017 10:59
Show Gist options
  • Star 6 You must be signed in to star a gist
  • Fork 2 You must be signed in to fork a gist
  • 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.
Download ZIP
devicehive-video-analysis DarkFlow modifications
Raw
build.py
# darkflow/net/build.py
def saveckpt(self):
self.saver.save(self.sess, './{}_model.ckpt'.format(self.meta['name']))
Raw
cli.py
# darkflow/cli.py
if FLAGS.saveckpt:
print('Rebuild a constant version ...')
tfnet.saveckpt(); exit('Done')
Raw
defaults.py
# 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