A neural network for identifying tags in E621 posts.
- MatplotLib >=
3.24
- Tensorflow >=
2.4
- Pandas >=
1.2.3
- Numpy >=
1.19.2
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.
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
.
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.