search

KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.
MIT License
2.65k stars 260 forks source link

about model #86

Open natsuonorepositories opened 1 year ago

natsuonorepositories commented 1 year ago

Hi!. I want to ask about model. I recognize model of deepdanbooru is used resnet isn't ok.

KichangKim commented 1 year ago

How it isnt okay? Can you provide more detail?

DeepDanbooru's model is based on resnet, but it is highly modified.

AlizerUncaged commented 1 year ago

@KichangKim is there a pre-trained model we can use?

EDIT: nevermind, I found automatic1111's pretrained model at https://github.com/AUTOMATIC1111/TorchDeepDanbooru/releases/download/v1/model-resnet_custom_v3.pt and im going to use it

Superfloh commented 1 year ago

This repo does have pre-trained models: https://github.com/KichangKim/DeepDanbooru/releases