diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-08-18 05:04:46 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-08-18 05:04:46 +0000 |
| commit | 8a1f32b6a5ce4dd5485fafd174eeeb142a877940 (patch) | |
| tree | 934dbaea48662f489df615b30ac39826eef49672 /modules/images.py | |
| parent | 541ef9247cfe8785c12f7142ad02cd5310fc5925 (diff) | |
| download | stable-diffusion-webui-gfx803-8a1f32b6a5ce4dd5485fafd174eeeb142a877940.tar.gz stable-diffusion-webui-gfx803-8a1f32b6a5ce4dd5485fafd174eeeb142a877940.tar.bz2 stable-diffusion-webui-gfx803-8a1f32b6a5ce4dd5485fafd174eeeb142a877940.zip | |
image hash
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/modules/images.py b/modules/images.py index 019c1d60..ea5df6dc 100644 --- a/modules/images.py +++ b/modules/images.py @@ -368,7 +368,8 @@ class FilenameGenerator: 'denoising': lambda self: self.p.denoising_strength if self.p and self.p.denoising_strength else NOTHING_AND_SKIP_PREVIOUS_TEXT,
'user': lambda self: self.p.user,
'vae_filename': lambda self: self.get_vae_filename(),
- 'none': lambda self: '', # Overrides the default so you can get just the sequence number
+ 'none': lambda self: '', # Overrides the default, so you can get just the sequence number
+ 'image_hash': lambda self, *args: self.image_hash(*args) # accepts formats: [image_hash<length>] default full hash
}
default_time_format = '%Y%m%d%H%M%S'
@@ -448,6 +449,10 @@ class FilenameGenerator: return sanitize_filename_part(formatted_time, replace_spaces=False)
+ def image_hash(self, *args):
+ length = int(args[0]) if (args and args[0] != "") else None
+ return hashlib.sha256(self.image.tobytes()).hexdigest()[0:length]
+
def apply(self, x):
res = ''
|