DanbooruClassifier

This repository contains code for my experiements with image classification on the Danbooru dataset, specifically danbooru2021.

1588 tag subset: 163,161,523 tags over 4,910,214 posts, ~157M for training (000-899) over ~4.42M posts

Name		Name	Last commit message	Last commit date
Latest commit History 1,288 Commits
.gitignore		.gitignore
DanbooruWikiEmbeddings1588.pth		DanbooruWikiEmbeddings1588.pth
DanbooruWikiEmbeddings1588_gte_large_en_v1.5_no_norm_d1024.pth		DanbooruWikiEmbeddings1588_gte_large_en_v1.5_no_norm_d1024.pth
DanbooruWikiEmbeddings5500.pth		DanbooruWikiEmbeddings5500.pth
DanbooruWikiEmbeddings5500_gte_large_en_v1.5_no_norm_d1024.pth		DanbooruWikiEmbeddings5500_gte_large_en_v1.5_no_norm_d1024.pth
LICENSE		LICENSE
README.md		README.md
build_cache.py		build_cache.py
danbooruDataset.py		danbooruDataset.py
evaluate.py		evaluate.py
handleMultiLabel.py		handleMultiLabel.py
infer.py		infer.py
modelZooCelebA.py		modelZooCelebA.py
models.py		models.py
parallelJsonReader.py		parallelJsonReader.py
tags1588.pkl		tags1588.pkl
tags5500.pkl		tags5500.pkl
trainer.py		trainer.py
trainer_CelebA.py		trainer_CelebA.py
trainer_tpu.py		trainer_tpu.py

Provide feedback