xsl
443cfa298f
包含： - hair_service_sd: 主服务（换发型/换发色/生发，端口8801） - photo_service: LoRA调度+训练（端口32678） - hair_grow_service: 调试测试页（端口8888，含4个测试页） - 批量训练脚本（batch_train_hairstyles.py） - 发际线mask自动识别（hairline_mask.py，4种方案） - 手绘mask换发型（hair_swap_manual.py） - 文档：README.md + LARGE_FILES.md + docs/ 大文件（模型权重200G、训练数据123G）已排除，见 LARGE_FILES.md OSS/COS密钥已脱敏为环境变量，原文件备份在本地
28 lines
946 B
Markdown
28 lines
946 B
Markdown
换发算法服务代码
|
||
|
||
相关配置：config/configure.ini \
|
||
模型:请存放于weights目录下 \
|
||
服务启动：执行 gunicorn server:app -c gunicorn_config.py \
|
||
测试相关代码请写在 unit_test.py里
|
||
|
||
建议环境：
|
||
python3.7，pytorch1.8,cu11.1,torchvision10.0 \
|
||
gunicorn需要在/usr/bin/gunicorn需要在里修改python地址
|
||
|
||
run run_copy_cost.py
|
||
|
||
|
||
|
||
需要启动的服务：
|
||
环境变量设置
|
||
export CRYPTOGRAPHY_OPENSSL_NO_LEGACY=1
|
||
1. 换发算法服务
|
||
cd /root/project/hair_service_sd
|
||
/usr/local/miniconda3/envs/condiff-train-hair/bin/python run_copy_cost_colorb64.py
|
||
2. webUI服务推理 + onediff
|
||
cd /root/project/onediff/stable-diffusion-webui
|
||
/usr/local/miniconda3/envs/onediff/bin/python python webui.py --api --listen --xformers --port 57860
|
||
./webui.sh --api --listen --disable-safe-unpickle --port 9038
|
||
3. photo_service
|
||
cd /root/project/photo_service
|
||
/usr/local/miniconda3/envs/py310/bin/python lora_train_service_1.py |