Model built using pretraining restnet50 model in pytorch, from the data collected using google-images-download. Also contains visualization techniques for the prediction using saliency maps.
Primary LanguageJupyter Notebook