alankbi/detecto_demo_4.py

Created February 11, 2020 01:48

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/9f58abf4134c33b48ef93fa3448ae97f.js"></script>
Save alankbi/9f58abf4134c33b48ef93fa3448ae97f to your computer and use it in GitHub Desktop.

Download ZIP

Train a custom model with Detecto in 4 lines of code

Raw

detecto_demo_4.py

	from detecto import core, utils, visualize

	dataset = core.Dataset('images/')
	model = core.Model(['alien', 'bat', 'witch'])

	model.fit(dataset)

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