Trypophobia Image Detector - Browser Plugin using Deep Learning
Ever wanted to censor inappropriate images on the web using deep neural networks?
The browser extension is available for Mozilla Firefox. You can try it here.
- Create a deep learning model for detecting trypophobia triggers suitable for running on CPU
- Create a plug and play browser plugin for censoring trypophobic images on the fly while browsing the internet running entirely client-side.
- Prepare a high quality data set for training trypophobia classifiers combining different data sources together
To do list
- One global browser-wide keras.js instance in the browser plugin, cache predictions based on image fingerprints, create a settings page
- Polish the browser plugin and
publish it in the plugin store(s)
The utilities contained in the
utils folder are small programs and scripts useful in generating the data set and easing the usage of the deep learning lab Neptune.
The data set
Download the data set v2 prepared in 2017-09
Note: The provided images may be or not be subjected to copyright. By downloading the data set you agree to use it only for research purposes.
- 6.5k trypophobia triggering images obtained from:
- 10.5k neutral images obtained from Google images using own scrapper:
- 10k by supplying it 5k randomly chosen words from this english dictionary and downloading 2 images per word
- 192 with bushes keyword (introduced in v2 to eliminate false positives for greenery)
- 181 with grass keyword (since v2)
- 98 with forest keyword (since v2)
Images have been divided into 4 folders
/valid/trypo- 500 random trypophobia triggering images
/valid/norm- 500 random neutral images
/train/trypo- rest trypophobia triggering images
/train/norm- rest neutral images
- Nonimage files and animated images have been removed using this our tool and this our tool.
- Downloaded images have been deduplicated using md5 hashes and fuzzy deduplication tool available in digiKam.
- Trypophobia triggers were manually checked and isolated from random spam using our tool.
- Images have been rescaled (maintaining aspect ratio) and cropped to 256x256 using our tool.
- Images have been split into train and validation sets using our tool.
Anyone interested in the "raw" unprocessed data please send us an email.
The browser plugin censors images encountered while browsing the web. It then uses supplied trained model to determine which images are safe to reveal and for which a warning must be issued. The extension is a WebExtension and was tested on Mozilla Firefox. Currently the extension works on most sites. You can try it here.