Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
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
You can’t perform that action at this time.