moonq/simple-labeler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

deal with empty data dir

Browse Source
This commit is contained in:
Ville Rantanen
2022-09-18 14:25:36 +03:00
parent aeca3ac5cb
commit af93f85763
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
code/labeler.py
View File

@@ -92,7 +92,7 @@ def get_current_image(images):
def get_image_list(): def get_image_list():
return sorted( image_list = sorted(
[ [
os.path.join(app.config["IMAGEDIR"], x) os.path.join(app.config["IMAGEDIR"], x)
for x in os.listdir(app.config["IMAGEDIR"]) for x in os.listdir(app.config["IMAGEDIR"])
@@ -100,6 +100,9 @@ def get_image_list():
], ],
key=natural_key, key=natural_key,
) )
if len(image_list) == 0:
return [None]
return image_list
def get_metadata(imagepath): def get_metadata(imagepath):
@@ -211,4 +214,4 @@ def show_image(id=None):
labels=labels, labels=labels,
id_minus=id_minus, id_minus=id_minus,
id_plus=id_plus, id_plus=id_plus,
) )