https://github.com/slyt4il/cunnydetect

Let AI decide whether your waifu is socially acceptable 💢💢
https://github.com/slyt4il/cunnydetect

anime cunny image-classification

Last synced: 3 days ago
JSON representation

Let AI decide whether your waifu is socially acceptable 💢💢

• Host: GitHub
• URL: https://github.com/slyt4il/cunnydetect
• Owner: Slyt4il
• Created: 2023-08-16T08:30:11.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2023-09-10T14:55:18.000Z (over 1 year ago)
• Last Synced: 2024-01-27T20:03:24.482Z (11 months ago)
• Topics: anime, cunny, image-classification
• Language: Python
• Homepage: https://slyt4il.github.io/CunnyDetect/
• Size: 73.8 MB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# CunnyDetect

CunnyDetect is a state-of-the-art image classifier running entirely in the browser that aims to solve the age-old conundrum: *"Is this thing a loli?"*

- Can detect the presence of loli imagery with sufficient accuracy if the weather is good. (Lower accuracy if it's raining outside or if you look at it funny.)

- Lightweight. Can run in the browser.

- Help convince your peers that your waifu is, in fact, not a loli.






    



 prediction: cunny 😭 



    



 prediction: not_cunny 😇 




## Model

| Model | Number of images | Validation accuracy |

| :---: | :--------------: | :----------------:  |

| mobilenetv3_finetuned(3M) | 26446 | 86.7 |

## Requirements

`TensorFlow 2.10.0` is required if you want GPU support on Windows. Also cudatoolkit, cuda-nvcc, and cudnn.

If not training, CPU should be enough for prediction.

## Usage

**Training**

Place images in `training_data`

```

python model.py

```

**Predicting**

Place images in `images_to_predict`

```

python predict.py

```

**Converting to tensorflowjs model**

```

tensorflowjs_converter --input_format=keras --output_format=tfjs_graph_model models/mobilenetv3_finetuned(3M).h5 client/js_model

```

## Training data

Images are sourced from gelbooru, of which many are too cursed to redistribute.

| Class | Example tags |

| :---- | :----------- |

| cunny | `1girl` `solo` `loli` `flat-chest` `female_focus` `highres`|

| not_cunny | `1girl` `1boy` `solo` `breasts` `mature_female` `female_focus` `male_focus` `highres` |