https://github.com/gorbak25/trypophobia-detector

https://github.com/gorbak25/trypophobia-detector

keras machine-learning trypophobia

Last synced: 29 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/gorbak25/trypophobia-detector
• Owner: gorbak25
• Created: 2017-01-16T13:38:30.000Z (about 8 years ago)
• Default Branch: summary
• Last Pushed: 2017-02-03T00:51:15.000Z (about 8 years ago)
• Last Synced: 2024-11-14T11:37:36.414Z (3 months ago)
• Topics: keras, machine-learning, trypophobia
• Language: Jupyter Notebook
• Size: 3.02 MB
• Stars: 2
• Watchers: 1
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# trypophobia-detector

This is a convolutional neural network which detects whether a given image contains trypophobia triggers(https://en.wikipedia.org/wiki/Trypophobia). 

This network was made during machine learning workshops in ICM UW from 2017-01-15 to 2017-01-20 with supervision of PhD Piotr Migdał (http://p.migdal.pl/).

The training and validation sets consisted of two classes - trypophobic and normal.

Using https://github.com/HoverHell/RedditImageGrab our team obtained trypophobic pictures from the tryphophobia subreddit and normal pictures from the img subreddit.

In each class there was the same ammount of pictures. The validation set contained 20% of the images.

This network obtained 88.44% accuracy on the validation set and used a pretrained vgg16 network. For more information about the vgg16 network see this paper:

Very Deep Convolutional Networks for Large-Scale Image Recognition K. Simonyan, A. Zisserman arXiv:1409.1556

The file training_process.ipynb contains detailed information about the training process.

The file load_model.ipynb contains a ready to use function to load this network.

The weights can be obtained at: https://drive.google.com/open?id=0B9fEN_YHQmF3cGJWNlE1Z1pJd1U