Skip to content

ALewdDev/E621-Tag-Labeler

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

E621 Tagger

A neural network for identifying tags in E621 posts.

Requirement

  • MatplotLib >= 3.24
  • Tensorflow >= 2.4
  • Pandas >= 1.2.3
  • Numpy >= 1.19.2

How it works

For the sake of being able to see the process of the architecture,
the model is only saved using weights, which also means it is
built during runtime and can be modified / loaded aftewards for
use in other projects.

Image Size

You can use any image size and disable the auto resizer (512px),
this is not advised however as it was trained on 128px, 256px and 512px.

Location & Format

By default the model uses png files, if you want to change this,
please edit Tagger.py and replace source_dir with the directory
and file type of your desire.

Languages

  • Python 100.0%