Bruno Madeira
657dd120e1
* patch_search.py renamed to synthesis_subroutines.py. * fixed number of channels when using cv.floodFill with latents in new min cut implementation. * make_seamless.py and make_seamless2.py use same "min_cut_patch" implementation. * quilting.py allows to use v0 implementation, but otherwise, also shares the same as seamless nodes. * check optional lib using importlib
31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
import importlib.util
|
|
from .nodes import *
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ImageQuilting_Bmad": ImageQuilting,
|
|
"LatentQuilting_Bmad": LatentQuilting,
|
|
"ImageQuiltingSeamlessMB_Bmad": ImageMakeSeamlessMB,
|
|
"LatentQuiltingSeamlessMB_Bmad": LatentMakeSeamlessMB,
|
|
"ImageQuiltingSeamlessSB_Bmad": ImageMakeSeamlessSB,
|
|
"LatentQuiltingSeamlessSB_Bmad": LatentMakeSeamlessSB,
|
|
"GuessQuiltingBlockSize_Bmad": GuessNiceBlockSize
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ImageQuilting_Bmad": "Image Quilting",
|
|
"LatentQuilting_Bmad": "Latent Quilting",
|
|
"ImageQuiltingSeamlessMB_Bmad": "Image Seamless Quilting MP",
|
|
"ImageQuiltingSeamlessSB_Bmad": "Image Seamless Quilting SP",
|
|
"LatentQuiltingSeamlessMB_Bmad": "Latent Seamless Quilting MP",
|
|
"LatentQuiltingSeamlessSB_Bmad": "Latent Seamless Quilting SP",
|
|
"GuessQuiltingBlockSize_Bmad": "Guess Quilting Block Size",
|
|
}
|
|
|
|
optional_package_name = 'pyastar2d'
|
|
if importlib.util.find_spec(optional_package_name) is None:
|
|
print(f"{optional_package_name} is not installed;"
|
|
f" jena2020 implementation will be used instead for min cut.")
|
|
else:
|
|
print(f"{optional_package_name} is installed;"
|
|
f" it will be used for computing the min cut unless specified otherwise.")
|