huggingface / controlnet_aux

Apache License 2.0
398 stars 86 forks source link

add Processor class #45

Closed ChristiaensBert closed 1 year ago

ChristiaensBert commented 1 year ago

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, ...)

ChristiaensBert commented 1 year ago

Also if you want, I could add the Segmentation model to the pre-processors

bertChristiaens1 commented 1 year ago

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 😁