Closed ChristiaensBert closed 1 year ago
Also if you want, I could add the Segmentation model to the pre-processors
Thanks for merging! However there still have to be some changes to make it fully functional with your code, I’ll make the changes and test them asap 😁
Hey Patrick Von Platen, I love your repo and I'm using it to create my own ControlNet datasets etc :D
I made a general Processor Class that makes it even easier to load in your different models. Currently I just copied the code from my processing pipeline, but let me know if you would like to add it to your repo, then I can clean it up, add some more functionality and integrate it will your folder structure.
You just need to pass the processor_id and it will load in the correct processor! I will still add functionality to put the model-based processors on CUDA etc and I can also change the way the user can choose the processor arguments (e.g. hand_and_face = True, ...)