let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lots of update 2

Browse Source
This commit is contained in:
Qing
2023-01-07 08:52:11 +08:00
parent a22536becc
commit a7240eedb5
13 changed files with 302 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lama_cleaner/file_manager/file_manager.py
View File

@@ -85,8 +85,9 @@ class FileManager:
names = sorted([it.name for it in glob_img(self.root_directory)])
res = []
for name in names:
img = Image.open(os.path.join(self.root_directory, name))
res.append({"name": name, "height": img.height, "width": img.width})
path = os.path.join(self.root_directory, name)
img = Image.open(path)
res.append({"name": name, "height": img.height, "width": img.width, "ctime": os.path.getctime(path)})
return res
@property

2
lama_cleaner/file_manager/utils.py
View File

@@ -48,7 +48,7 @@ def aspect_to_string(size):
return "x".join(map(str, size))
IMG_SUFFIX = {'.jpg', '.jpeg', '.png'}
IMG_SUFFIX = {'.jpg', '.jpeg', '.png', '.JPG', '.JPEG', '.PNG'}
def glob_img(p: Union[Path, str], recursive: bool = False):