search

Particle1904 / DatasetHelpers

Dataset Helper program to automatically select, re scale and tag Datasets (composed of image and text) for Machine Learning training.
MIT License
157 stars 9 forks source link

[REQUEST} Custom WD tags #35

Closed BenDes21 closed 1 month ago

BenDes21 commented 1 month ago

Hi there! Sorry to post in the issues section but I don't see the discussion tab.

Would like to know if it's possible to allow the user to use specific WD taggers ( like the ones from smiling wolf )

Thanks a lot for your amazing tool

Particle1904 commented 1 month ago

You can download the .onnx model and and the .csv and rename it to the file names the application uses but I wont offer any support.

All the SmillingWolf's auto taggers have the exact same input and outputs, so they are all interchangeable. As long as the model have a .onnx version, you can use any of them by renaming the files. Rename the model to "wdV3Model.onnx" and the corresponding .csv file to "wdV3Tags.csv" and put them inside the "models" folder inside the application root folder. Are you asking because you want to use the new "wd-eva02-large-tagger-v3" model? If yes, I'll add support for it today. It will be available in v2.9.0 version.

Edit: I've opened the issue again to clarify.

BenDes21 commented 1 month ago

Are you asking because you want to use the new "wd-eva02-large-tagger-v3" model? If yes, I'll add support for it today. It will be available in v2.9.0 version.

Hi there ! Thanks a lot! Yes my request was for wd-eva02-large-tagger-v3 and wd-vit-large-tagger-v3 :)

Have a great day

Particle1904 commented 1 month ago

Here's the release with the large model: https://github.com/Particle1904/DatasetHelpers/releases/tag/v2.9.0 Let me know if you find any issues with it.