alankbi/detecto_demo_8.py

Created February 14, 2020 17:39

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/alankbi/c1d83b3d733cefd96ed45bd66e54d725.js"></script>
Save alankbi/c1d83b3d733cefd96ed45bd66e54d725 to your computer and use it in GitHub Desktop.

Download ZIP

Save and load models from files with Detecto

Raw

detecto_demo_8.py

	model.save('model_weights.pth')

	# ... Later ...

	model = core.Model.load('model_weights.pth', ['alien', 'bat', 'witch'])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment