Question

I'd like to get started with machine learning, specifically image recognition. I know that Python is the most popular language for ML since it's easy to pick up and there's tons of libraries for it. I have a few questions:

  1. What are the advantages of using a pre-trained model vs training my own?
  2. Where would I find pre-trained models? At what point would it be worth it to train my own models, and how can I do so?
  3. Which library should I be using as a beginner? TensorFlow, scikit-learn, etc.
Was it helpful?

Solution

What are the advantages of using a pre-trained model vs training my own?

The main advantage of pre-trained models is that it greatly reduces required data for your model. You can see it as a better way to initialize your model.

Where would I find pre-trained models? At what point would it be worth it to train my own models, and how can I do so?

Keras has a variety of pretrained models for image classification. If your data consists of rare objects or uncommon situations, it might be better to train your own model, say for very specific task. Pretrained models only work if patterns in your data share similarities to patterns learned by the pretrained model.

Which library should I be using as a beginner? TensorFlow, scikit-learn, etc.

Scikit-learn and Keras are pretty straightforward. They both have toy datasets for experimenting. Deep Learning with Python by François Chollet has a nice chapter on computer vision, which has a section on using pretrained models.

Licensed under: CC-BY-SA with attribution
Not affiliated with datascience.stackexchange
scroll top