colomi
0eb61f3e60
包含： - 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 排除， 由网盘单独上传。
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import os
|
|
|
|
if __name__ == '__main__':
|
|
in_dir = "/mnt/database2/jiangqian/0808/online_orig_train_datas_2"
|
|
out_dir = "/mnt/database2/jiangqian/0808/online_train_datas_2"
|
|
|
|
dirs = os.listdir(in_dir)
|
|
|
|
for single_dir in dirs:
|
|
if "traindata" in single_dir:
|
|
in_single_dir = os.path.join(in_dir, single_dir)
|
|
out_single_dir = os.path.join(out_dir, single_dir)
|
|
|
|
if not os.path.exists(out_single_dir):
|
|
os.makedirs(out_single_dir)
|
|
|
|
images_dir = os.path.join(out_single_dir, "images")
|
|
if not os.path.exists(images_dir):
|
|
os.makedirs(images_dir)
|
|
|
|
hairstyle_dir = os.path.join(images_dir, "1_hairstyle")
|
|
if not os.path.exists(hairstyle_dir):
|
|
os.makedirs(hairstyle_dir)
|
|
|
|
for root, dirs, files in os.walk(in_single_dir):
|
|
for file in files:
|
|
in_file = os.path.join(root, file)
|
|
out_file = os.path.join(hairstyle_dir, file)
|
|
os.system("cp %s %s" % (in_file, out_file))
|
|
print("copy %s to %s" % (in_file, out_file))
|
|
|
|
|