nodes/mcmonkeyprojects-sd-dynamic-thresholding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor code cleanup

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-10-29 07:20:15 -07:00
parent 9ed3eb7f89
commit 450983c621
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dynthres_core.py
+16 -15
View File
@@ -8,12 +8,12 @@ class DynThresh:
Startpoints = ["MEAN", "ZERO"] Startpoints = ["MEAN", "ZERO"]
Variabilities = ["AD", "STD"] Variabilities = ["AD", "STD"]
def __init__(self, mimic_scale, threshold_percentile, mimic_mode, mimic_scale_min, cfg_mode, cfg_scale_min, sched_val, experiment_mode, maxSteps, separate_feature_channels, scaling_startpoint, variability_measure, interpolate_phi): def __init__(self, mimic_scale, threshold_percentile, mimic_mode, mimic_scale_min, cfg_mode, cfg_scale_min, sched_val, experiment_mode, max_steps, separate_feature_channels, scaling_startpoint, variability_measure, interpolate_phi):
self.mimic_scale = mimic_scale self.mimic_scale = mimic_scale
self.threshold_percentile = threshold_percentile self.threshold_percentile = threshold_percentile
self.mimic_mode = mimic_mode self.mimic_mode = mimic_mode
self.cfg_mode = cfg_mode self.cfg_mode = cfg_mode
self.maxSteps = maxSteps self.max_steps = max_steps
self.cfg_scale_min = cfg_scale_min self.cfg_scale_min = cfg_scale_min
self.mimic_scale_min = mimic_scale_min self.mimic_scale_min = mimic_scale_min
self.experiment_mode = experiment_mode self.experiment_mode = experiment_mode
@@ -25,32 +25,33 @@ class DynThresh:
def interpretScale(self, scale, mode, min): def interpretScale(self, scale, mode, min):
scale -= min scale -= min
max = self.maxSteps - 1 max = self.max_steps - 1
frac = self.step / max
if mode == "Constant": if mode == "Constant":
pass pass
elif mode == "Linear Down": elif mode == "Linear Down":
scale *= 1.0 - (self.step / max) scale *= 1.0 - frac
elif mode == "Half Cosine Down": elif mode == "Half Cosine Down":
scale *= math.cos((self.step / max)) scale *= math.cos(frac)
elif mode == "Cosine Down": elif mode == "Cosine Down":
scale *= math.cos((self.step / max) * 1.5707) scale *= math.cos(frac * 1.5707)
elif mode == "Linear Up": elif mode == "Linear Up":
scale *= self.step / max scale *= frac
elif mode == "Half Cosine Up": elif mode == "Half Cosine Up":
scale *= 1.0 - math.cos((self.step / max)) scale *= 1.0 - math.cos(frac)
elif mode == "Cosine Up": elif mode == "Cosine Up":
scale *= 1.0 - math.cos((self.step / max) * 1.5707) scale *= 1.0 - math.cos(frac * 1.5707)
elif mode == "Power Up": elif mode == "Power Up":
scale *= math.pow(self.step / max, self.sched_val) scale *= math.pow(frac, self.sched_val)
elif mode == "Power Down": elif mode == "Power Down":
scale *= 1.0 - math.pow(self.step / max, self.sched_val) scale *= 1.0 - math.pow(frac, self.sched_val)
elif mode == "Linear Repeating": elif mode == "Linear Repeating":
portion = ((self.step / max) * self.sched_val) % 1.0 portion = (frac * self.sched_val) % 1.0
scale *= (0.5 - portion) * 2 if portion < 0.5 else (portion - 0.5) * 2 scale *= (0.5 - portion) * 2 if portion < 0.5 else (portion - 0.5) * 2
elif mode == "Cosine Repeating": elif mode == "Cosine Repeating":
scale *= math.cos((self.step / max) * 6.28318 * self.sched_val) * 0.5 + 0.5 scale *= math.cos(frac * 6.28318 * self.sched_val) * 0.5 + 0.5
elif mode == "Sawtooth": elif mode == "Sawtooth":
scale *= ((self.step / max) * self.sched_val) % 1.0 scale *= (frac * self.sched_val) % 1.0
scale += min scale += min
return scale return scale
@@ -155,7 +156,7 @@ class DynThresh:
maxR, maxG, maxB, maxW = resRGB[0][0].max(), resRGB[0][1].max(), resRGB[0][2].max(), resRGB[0][3].max() maxR, maxG, maxB, maxW = resRGB[0][0].max(), resRGB[0][1].max(), resRGB[0][2].max(), resRGB[0][3].max()
maxRGB = max(maxR, maxG, maxB) maxRGB = max(maxR, maxG, maxB)
print(f"test max = r={maxR}, g={maxG}, b={maxB}, w={maxW}, rgb={maxRGB}") print(f"test max = r={maxR}, g={maxG}, b={maxB}, w={maxW}, rgb={maxRGB}")
if self.step / (self.maxSteps - 1) > 0.2: if self.step / (self.max_steps - 1) > 0.2:
if maxRGB < 2.0 and maxW < 3.0: if maxRGB < 2.0 and maxW < 3.0:
resRGB /= maxRGB / 2.4 resRGB /= maxRGB / 2.4
else: else: