Skip to content

Instantly share code, notes, and snippets.

@sugartom
Forked from igor-panteleev/build.py
Created July 2, 2018 18:13
Show Gist options
  • Save sugartom/2c38cfb6816681f151e63739cf54d95a to your computer and use it in GitHub Desktop.
Save sugartom/2c38cfb6816681f151e63739cf54d95a 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