/Weeb-Detector

An AI taught to recognize weebs in discord profile pictures

Primary LanguagePythonMIT LicenseMIT

GitHub

Weeb-Detector

An AI taught to recognize weebs in discord profile pictures. It takes an image and then outputs how certain it is that the image contained anime. I personally collected and labeled close to 1000 discord profile pictures from random servers to use as training material. Unfortunately due to github restrictions it had to be uploaded in a .zip file in the Dataset directory. The checkpoint of the model couldn't be included for the same reason. The AI trained from the database might have weird behavior on non-square images or images with high resolution and detail

Examples:

Weeb:

test_weeb

2022-11-06_19-01_1

Normal:

test_normal

2022-11-06_19-01

How to use:

  1. Unzip the database
  2. Run the DetectorTrainer.py script to train a model(It shouldn't take longer than half an hour on a potato PC)
  3. Run TestImage.py (Replace the image path with a path to your own image)

Do what you want with the code, but credit would be much appreciated. Contact info in the profile