初始化换发型项目：3个微服务代码 + 部署脚本

包含： - hair_service_sd: 换发型/换发色算法服务 (端口 8801) - photo_service: LoRA 训练调度服务 (端口 32678) - stable-diffusion-webui: SD WebUI 推理服务 (端口 57860) - kohya_ss_home: 训练环境代码 - meidaojia: 监控测试脚本 - setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查) - start_all_services.sh: 启动3个服务 - configure.ini.template: 路径模板化 (BASE_DIR自动推导) - conda_envs/py310.yml: py310 环境定义大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除，由网盘单独上传。
2026-07-11 18:11:49 +08:00
commit 0eb61f3e60
628 changed files with 120882 additions and 0 deletions
@@ -0,0 +1,106 @@
+import io
+import cv2
+import base64
+import requests
+from PIL import Image
+import numpy as np
+
+"""
+    To use this example make sure you've done the following steps before executing:
+    1. Ensure automatic1111 is running in api mode with the controlnet extension. 
+       Use the following command in your terminal to activate:
+            ./webui.sh --no-half --api
+    2. Validate python environment meet package dependencies.
+       If running in a local repo you'll likely need to pip install cv2, requests and PIL 
+"""
+
+
+class ControlnetRequestImg2Img:
+    def __init__(self, prompt, net_prompt):
+        self.url = "http://127.0.0.1:7860/sdapi/v1/img2img"
+        self.prompt = prompt
+        self.neg_prompt = net_prompt
+        self.body = None
+
+    def build_body(self, dst_width, dst_height, cfg_scale, base_img):
+
+        self.body = {
+            "prompt": self.prompt,
+            "negative_prompt": self.neg_prompt,
+            "sampler_name": "Restart",
+            "batch_size": 1,
+            "steps": 30,
+            "width": dst_width,
+            "height": dst_height,
+            "cfg_scale": cfg_scale,
+            "seed": -1,
+            "init_images": [
+                self.encode_image_to_base64(base_img)
+            ],
+            "denoising_strength": 0.4,
+            "alwayson_scripts": {
+                "controlnet": {
+                    "args": [
+                        {
+                            "enabled": True,
+                            "module": "openpose_full",
+                            "model": "openpose",
+                            "weight": 1.0,
+                            # "image": self.read_image(),
+                            "resize_mode": "Crop and Resize",
+                            "low_vram": False,
+                            "processor_res": 512,
+                            "guidance_start": 0.0,
+                            "guidance_end": 1.0,
+                            "control_mode": "Balanced",
+                            "pixel_perfect": True
+                        }
+                    ]
+                }
+            }
+        }
+
+    def send_request(self):
+        response = requests.post(url=self.url, json=self.body)
+        return response.json()
+
+    def encode_image_to_base64(self, img):
+        retval, bytes = cv2.imencode('.png', img)
+        encoded_image = base64.b64encode(bytes).decode('utf-8')
+        return encoded_image
+
+    def read_image(self):
+        img = cv2.imread(self.img_path)
+        retval, bytes = cv2.imencode('.png', img)
+        encoded_image = base64.b64encode(bytes).decode('utf-8')
+        return encoded_image
+    
+    def read_mask(self):
+        img = cv2.imread(self.mask)
+        retval, bytes = cv2.imencode('.png', img)
+        encoded_image = base64.b64encode(bytes).decode('utf-8')
+        return encoded_image
+
+def encode_image_to_base64(img):
+    retval, bytes = cv2.imencode('.jpg', img)
+    encoded_image = base64.b64encode(bytes).decode('utf-8')
+    return encoded_image
+
+
+
+if __name__ == '__main__':
+    path = '/home/chinatszrn/Downloads/photo_service/service_data/template_data/template01.png'
+    img = cv2.imread(path)
+    prompt = '<lora:5b05d5eeee0188f436d7131c4f0ff52b:0.8>,easyphoto_face, easyphoto, 1person,face,suit'
+    neg_prompt = '(worst quality:2),(low quality:2),(normal quality:2),lowres,watermark'
+
+
+    control_net = ControlnetRequestImg2Img(prompt, neg_prompt)
+    control_net.build_body(dst_width=img.shape[1], dst_height=img.shape[0], cfg_scale=3.5, base_img=img)
+    output = control_net.send_request()
+    result = output['images'][0]
+
+    image_array = np.frombuffer(base64.b64decode(result.split(",", 1)[0]), np.uint8)
+    image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
+    cv2.imshow('image', image)
+    cv2.waitKey()