© Thimira Amaratunga 2021
T. AmaratungaDeep Learning on Windowshttps://doi.org/10.1007/978-1-4842-6431-7_7

7. Transfer Learning

Thimira Amaratunga1  
(1)
Nugegoda, Sri Lanka
 

We saw how exceptionally well deep learning models performed when applied to computer vision and classification tasks. Our LeNet model with the MNIST and Fashion-MNIST datasets was able to achieve 90%–99% accuracy under a very reasonable amount of training time. We have also seen how the ImageNet models have achieved record-breaking accuracy levels in more complex datasets.

Now you might be eager to try out what we learned on a more complex and practical classification task. But what should we consider when we are going to train our own image classification model with our own categories ...

Get Deep Learning on Windows: Building Deep Learning Computer Vision Systems on Microsoft Windows now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.