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密钥已脱敏为环境变量，原文件备份在本地
102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# @File : setup.py
|
|
# @Time : 2020/1/15
|
|
# @Author : yangchaojie (yangchaojie@immomo.com)
|
|
|
|
import os
|
|
import sys
|
|
import shutil
|
|
import numpy
|
|
import tempfile
|
|
|
|
from setuptools import setup
|
|
from setuptools.extension import Extension
|
|
|
|
from Cython.Build import cythonize
|
|
from Cython.Distutils import build_ext
|
|
|
|
import platform
|
|
|
|
|
|
def get_root_path(root):
|
|
if os.path.dirname(root) in ['', '.']:
|
|
return os.path.basename(root)
|
|
else:
|
|
return get_root_path(os.path.dirname(root))
|
|
|
|
|
|
def copy_file(src, dest):
|
|
if os.path.exists(dest):
|
|
return
|
|
|
|
if not os.path.exists(os.path.dirname(dest)):
|
|
os.makedirs(os.path.dirname(dest))
|
|
if os.path.isdir(src):
|
|
shutil.copytree(src, dest)
|
|
else:
|
|
shutil.copyfile(src, dest)
|
|
|
|
|
|
def touch_init_file():
|
|
init_file_name = os.path.join(tempfile.mkdtemp(), '__init__.py')
|
|
with open(init_file_name, 'w'):
|
|
pass
|
|
return init_file_name
|
|
|
|
|
|
|
|
|
|
def compose_extensions(root='.'):
|
|
for file_ in os.listdir(root):
|
|
abs_file = os.path.join(root, file_)
|
|
|
|
if os.path.isfile(abs_file):
|
|
if abs_file.endswith('.py'):
|
|
extensions.append(Extension(get_root_path(abs_file) + '.*', [abs_file]))
|
|
elif abs_file.endswith('.c') or abs_file.endswith('.pyc'):
|
|
continue
|
|
else:
|
|
copy_file(abs_file, os.path.join(build_root_dir, abs_file))
|
|
if abs_file.endswith('__init__.py'):
|
|
copy_file(init_file, os.path.join(build_root_dir, abs_file))
|
|
|
|
else:
|
|
if os.path.basename(abs_file) in ignore_folders :
|
|
continue
|
|
if os.path.basename(abs_file) in conf_folders:
|
|
copy_file(abs_file, os.path.join(build_root_dir, abs_file))
|
|
compose_extensions(abs_file)
|
|
|
|
|
|
|
|
# if __name__ == '__main__':
|
|
build_root_dir = 'build/lib.' + platform.system().lower() + '-' + platform.machine() + '-' + str(
|
|
sys.version_info.major) + '.' + str(sys.version_info.minor)
|
|
|
|
print(build_root_dir)
|
|
|
|
extensions = []
|
|
ignore_folders = ['build', 'new_ref_zao_color_0818', 'ref_hair_online_0703_local', 'ref_分好类别', '.git']
|
|
conf_folders = ['conf']
|
|
|
|
|
|
init_file = touch_init_file()
|
|
print(init_file)
|
|
|
|
|
|
compose_extensions()
|
|
os.remove(init_file)
|
|
|
|
setup(
|
|
name='moxie_hairstyle',
|
|
version='1.0',
|
|
ext_modules=cythonize(
|
|
extensions,
|
|
nthreads=16,
|
|
compiler_directives=dict(always_allow_keywords=True),
|
|
include_path=[numpy.get_include()]),
|
|
cmdclass=dict(build_ext=build_ext))
|
|
|
|
# python setup.py build_ext |