Spaces:
Runtime error
Runtime error
File size: 4,085 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
# This is a python script to convert all old preprocessors to new format.
# However, the old preprocessors are not very memory effective
# and eventually we should move all old preprocessors to new format manually
# see also the forge_preprocessor_normalbae/scripts/preprocessor_normalbae for
# how to make better implementation of preprocessors.
# No newer preprocessors should be written in this legacy way.
# Never add new leagcy preprocessors please.
# The new forge_preprocessor_normalbae/scripts/preprocessor_normalbae
# is much more effective and maintainable
from annotator.util import HWC3
from .preprocessor_compiled import legacy_preprocessors
from ...supported_preprocessor import Preprocessor, PreprocessorParameter
###
# This file has lots of unreasonable historical designs and should be viewed as a frozen blackbox library.
# If you want to add preprocessor,
# please instead look at `extensions-builtin/forge_preprocessor_normalbae/scripts/preprocessor_normalbae`
# If you want to use preprocessor,
# please instead use `from modules_forge.shared import supported_preprocessors`
# and then use any preprocessor like: depth_midas = supported_preprocessors['depth_midas']
# Please do not hack/edit/modify/rely-on any codes in this file.
# Never use methods in this file to add anything!
# This file will be eventually removed but the workload is super high and we need more time to do this.
###
class LegacyPreprocessor(Preprocessor):
def __init__(self, name: str, legacy_dict):
super().__init__(name)
self._label = legacy_dict["label"]
self.call_function = legacy_dict["call_function"]
self.unload_function = legacy_dict["unload_function"]
self.managed_model = legacy_dict["managed_model"]
self.do_not_need_model = legacy_dict["model_free"]
self.show_control_mode = not legacy_dict["no_control_mode"]
self.sorting_priority = legacy_dict["priority"]
self.tags = legacy_dict["tags"]
self.returns_image = legacy_dict.get("returns_image", True)
if legacy_dict.get("use_soft_projection_in_hr_fix", False):
self.use_soft_projection_in_hr_fix = True
if legacy_dict["resolution"] is None:
self.resolution = PreprocessorParameter(visible=False)
else:
legacy_dict["resolution"]["label"] = "Resolution"
legacy_dict["resolution"]["step"] = 8
self.resolution = PreprocessorParameter(
**legacy_dict["resolution"], visible=True
)
if legacy_dict["slider_1"] is None:
self.slider_1 = PreprocessorParameter(visible=False)
else:
self.slider_1 = PreprocessorParameter(
**legacy_dict["slider_1"], visible=True
)
if legacy_dict["slider_2"] is None:
self.slider_2 = PreprocessorParameter(visible=False)
else:
self.slider_2 = PreprocessorParameter(
**legacy_dict["slider_2"], visible=True
)
if legacy_dict["slider_3"] is None:
self.slider_3 = PreprocessorParameter(visible=False)
else:
self.slider_3 = PreprocessorParameter(
**legacy_dict["slider_3"], visible=True
)
def __call__(
self,
input_image,
resolution,
slider_1=None,
slider_2=None,
slider_3=None,
**kwargs
):
# Legacy Preprocessors does not have slider 3
del slider_3
if self.managed_model is not None:
assert self.unload_function is not None
result, is_image = self.call_function(
img=input_image, res=resolution, thr_a=slider_1, thr_b=slider_2, **kwargs
)
if is_image and "inpaint" not in self.name:
result = HWC3(result)
if self.unload_function is not None:
self.unload_function()
return result
for name, data in legacy_preprocessors.items():
p = LegacyPreprocessor(name, data)
Preprocessor.add_supported_preprocessor(p)
|