chore: 适配本机 xsl 路径(RTX 5090 机器)
将所有绝对路径从 /home/ubuntu 改回 /home/xsl,覆盖启动脚本、 configure.ini 及训练/测试脚本。纯路径替换,无功能性改动。 本分支(xsl5090)用于本机(RTX 5090)运行;master 维持 /home/ubuntu 供远程 ubuntu 机器使用。
This commit is contained in:
@@ -6,8 +6,8 @@
|
||||
串接 prepare_train_data / gen_template_material / photo_service训练 / 回调 / 预览图
|
||||
|
||||
用法:
|
||||
cd /home/ubuntu/change_hair/project/hair_service_sd
|
||||
python /home/ubuntu/change_hair/train_hairstyle_full.py --hair-id huaban1 --input /home/ubuntu/change_hair/huaban1 --gender girl --template-img /home/ubuntu/change_hair/huaban1/huaban1.jpg
|
||||
cd /home/xsl/change_hair/project/hair_service_sd
|
||||
python /home/xsl/change_hair/train_hairstyle_full.py --hair-id huaban1 --input /home/xsl/change_hair/huaban1 --gender girl --template-img /home/xsl/change_hair/huaban1/huaban1.jpg
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
@@ -21,7 +21,7 @@ import argparse
|
||||
import requests as req
|
||||
|
||||
ssl._create_default_https_context = ssl._create_unverified_context
|
||||
HAIR_SERVICE_DIR = "/home/ubuntu/change_hair/project/hair_service_sd"
|
||||
HAIR_SERVICE_DIR = "/home/xsl/change_hair/project/hair_service_sd"
|
||||
os.chdir(HAIR_SERVICE_DIR)
|
||||
sys.path.insert(0, HAIR_SERVICE_DIR)
|
||||
os.environ.setdefault("HF_HUB_OFFLINE", "1")
|
||||
@@ -48,9 +48,9 @@ RESOLUTIONS = [512, 768, 1024, 1280, 1536]
|
||||
PHOTO_TRAIN = "http://127.0.0.1:32678/api/hair/train"
|
||||
HAIR_CALLBACK = "http://127.0.0.1:8801/api/hair/trainCallBack"
|
||||
SWAP_API = "http://127.0.0.1:8801/api/swapHair/v1"
|
||||
GIRL_IMG = "/home/ubuntu/change_hair/images/girl.png"
|
||||
BOY_IMG = "/home/ubuntu/change_hair/images/boy.png"
|
||||
PREVIEW_DIR = "/home/ubuntu/change_hair/hair_grow_service/static/previews"
|
||||
GIRL_IMG = "/home/xsl/change_hair/images/girl.png"
|
||||
BOY_IMG = "/home/xsl/change_hair/images/boy.png"
|
||||
PREVIEW_DIR = "/home/xsl/change_hair/hair_grow_service/static/previews"
|
||||
|
||||
_detector = _aligner = _matte = None
|
||||
def get_models():
|
||||
|
||||
Reference in New Issue
Block a user