Skip to content

iammartian0/computer-vision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

This is edited notebook from the course Practical deeplearning for coders.

The main goal of this notebook is to take advantage of capabilities of existing pretrained models. To conclude in a single word - 'Transfer Learning'

The model used is pretrained 'Resnet18', finetuned on images of target classes. The images are scrapped from internet.

The model took around 3 minutes to train on kaggle 'CPU'.

The inference app is hosted on Huggingface Space.