https://github.com/apppear/assignment2
PPKE ITK Neural Network assignment 2
https://github.com/apppear/assignment2
Last synced: over 1 year ago
JSON representation
PPKE ITK Neural Network assignment 2
- Host: GitHub
- URL: https://github.com/apppear/assignment2
- Owner: AppPear
- Created: 2017-12-30T14:31:15.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2018-01-13T12:58:03.000Z (over 8 years ago)
- Last Synced: 2025-01-21T00:20:12.421Z (over 1 year ago)
- Language: Jupyter Notebook
- Homepage:
- Size: 5.9 MB
- Stars: 0
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Assignment 2 - transfer learning
Class project for Neural network class. I classified the following classes:
* Big Ben
* Boris bus (the red bus in london)
* Shard
* Tube sign
I collected my learning set via google images, I used a chorme extension called bulk image downloader, which made it a lot faster.
I was abel to achive pretty good results in a short time, training took 1 min 12 sec and the accuracy is around 97%
Weights can be downloaded from here: https://www.dropbox.com/s/y17251zl41i1ae0/assignment2_weights.pt?dl=0
I followed the tutorial found on pytorch website and also I found another great tutorial which helped me a lot ( http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html, https://towardsdatascience.com/transfer-learning-using-pytorch-part-2-9c5b18e15551 )