1580158
/
stable-diffusion-webui-dataset-tag-editor
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix holding images on RAM when move or delete (#39)

pull/41/head
toshiaki1729 2023-02-18 16:19:36 +09:00
parent 6b1c76849a
commit 7123254545
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/dataset_tag_editor/dataset_tag_editor.py
View File

@ -392,6 +392,8 @@ class DatasetTagEditor:
if delete_image:
try:
if os.path.exists(img_path) and os.path.isfile(img_path):
if img_path in self.images:
del self.images[img_path]
os.remove(img_path)
self.dataset.remove_by_path(img_path)
print(f'[tag-editor] Deleted {img_path}')
@ -431,8 +433,12 @@ class DatasetTagEditor:
try:
dst_path = os.path.join(dest_dir, os.path.basename(img_path))
if os.path.exists(img_path) and os.path.isfile(img_path):
if img_path in self.images:
img = self.images[img_path]
del self.images[img_path]
os.replace(img_path, dst_path)
self.dataset.remove_by_path(img_path)
self.images[dst_path] = img
print(f'[tag-editor] Moved {img_path} -> {dst_path}')
except Exception as e:
print(e)