|
import os |
|
import platform |
|
|
|
import requests |
|
|
|
|
|
def build_ebsynth(): |
|
if os.path.exists('src/ebsynth/deps/ebsynth/bin/ebsynth'): |
|
print('Ebsynth has been built.') |
|
return |
|
|
|
os_str = platform.system() |
|
|
|
if os_str == 'Windows': |
|
print('Build Ebsynth Windows 64 bit.', |
|
'If you want to build for 32 bit, please modify install.py.') |
|
cmd = '.\\build-win64-cpu+cuda.bat' |
|
exe_file = 'src/ebsynth/deps/ebsynth/bin/ebsynth.exe' |
|
elif os_str == 'Linux': |
|
cmd = 'bash build-linux-cpu+cuda.sh' |
|
exe_file = 'src/ebsynth/deps/ebsynth/bin/ebsynth' |
|
elif os_str == 'Darwin': |
|
cmd = 'sh build-macos-cpu_only.sh' |
|
exe_file = 'src/ebsynth/deps/ebsynth/bin/ebsynth.app' |
|
else: |
|
print('Cannot recognize OS. Ebsynth installation stopped.') |
|
return |
|
|
|
os.chdir('src/ebsynth/deps/ebsynth') |
|
print(cmd) |
|
os.system(cmd) |
|
os.chdir('../../../..') |
|
if os.path.exists(exe_file): |
|
print('Ebsynth installed successfully.') |
|
else: |
|
print('Failed to install Ebsynth.') |
|
|
|
|
|
def download(url, dir, name=None): |
|
os.makedirs(dir, exist_ok=True) |
|
if name is None: |
|
name = url.split('/')[-1] |
|
path = os.path.join(dir, name) |
|
if not os.path.exists(path): |
|
print(f'Install {name} ...') |
|
open(path, 'wb').write(requests.get(url).content) |
|
print('Install successfully.') |
|
|
|
|
|
def download_gmflow_ckpt(): |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/gmflow_sintel-0c07dcb3.pth') |
|
download(url, 'model') |
|
|
|
|
|
def download_egnet_ckpt(): |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/epoch_resnet.pth') |
|
download(url, 'model') |
|
|
|
def download_hed_ckpt(): |
|
url = ('https://huggingface.co/lllyasviel/Annotators/' |
|
'resolve/main/ControlNetHED.pth') |
|
download(url, 'src/ControlNet/annotator/ckpts') |
|
|
|
def download_depth_ckpt(): |
|
url = ('https://huggingface.co/lllyasviel/ControlNet/' |
|
'resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt') |
|
download(url, 'src/ControlNet/annotator/ckpts') |
|
|
|
def download_ebsynth_ckpt(): |
|
os_str = platform.system() |
|
if os_str == 'Linux': |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/ebsynth') |
|
download(url, 'src/ebsynth/deps/ebsynth/bin') |
|
elif os_str == 'Windows': |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/ebsynth.exe') |
|
download(url, 'src/ebsynth/deps/ebsynth/bin') |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/ebsynth_cpu.dll') |
|
download(url, 'src/ebsynth/deps/ebsynth/bin') |
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/' |
|
'resolve/main/models/ebsynth_cpu.exe') |
|
download(url, 'src/ebsynth/deps/ebsynth/bin') |
|
else: |
|
print('No available compiled Ebsynth.') |
|
|
|
|
|
download_ebsynth_ckpt() |
|
download_gmflow_ckpt() |
|
download_egnet_ckpt() |
|
download_hed_ckpt() |
|
download_depth_ckpt() |
|
|
