| from backend.annotators.canny_control import CannyControl |
| from backend.annotators.depth_control import DepthControl |
| from backend.annotators.lineart_control import LineArtControl |
| from backend.annotators.mlsd_control import MlsdControl |
| from backend.annotators.normal_control import NormalControl |
| from backend.annotators.pose_control import PoseControl |
| from backend.annotators.shuffle_control import ShuffleControl |
| from backend.annotators.softedge_control import SoftEdgeControl |
|
|
|
|
| class ImageControlFactory: |
| def create_control(self, controlnet_type: str): |
| if controlnet_type == "Canny": |
| return CannyControl() |
| elif controlnet_type == "Pose": |
| return PoseControl() |
| elif controlnet_type == "MLSD": |
| return MlsdControl() |
| elif controlnet_type == "Depth": |
| return DepthControl() |
| elif controlnet_type == "LineArt": |
| return LineArtControl() |
| elif controlnet_type == "Shuffle": |
| return ShuffleControl() |
| elif controlnet_type == "NormalBAE": |
| return NormalControl() |
| elif controlnet_type == "SoftEdge": |
| return SoftEdgeControl() |
| else: |
| print("Error: Control type not implemented!") |
| raise Exception("Error: Control type not implemented!") |
|
|
