adapt download folder to image dataset style

app.py

@@ -11,8 +11,8 @@ from flask import send_file
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 
-UPLOAD_FOLDER = './images'
-OUT_FOLDER = './anno'
+UPLOAD_FOLDER = 'images'
+OUT_FOLDER = 'anno'
 
 app = Flask(__name__)
 app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0
@@ -20,14 +20,14 @@ app.config["SECRET_KEY"] = '#thaistartsfirst!'
 app.config["IMAGES"] = UPLOAD_FOLDER
 app.config["OUT"] = OUT_FOLDER
 app.config["LABELS"] = []
-app.config["CATEGORIES"] = {} # []
+app.config["CATEGORIES"] = {}  # []
 app.config["HEAD"] = 0
 app.config["SESSION_PERMANENT"] = False
 
 # Ensure the upload directory exists
-if not os.path.exists(UPLOAD_FOLDER):
+if not os.path.exists(os.path.join(os.getcwd(), UPLOAD_FOLDER)):
     os.makedirs(UPLOAD_FOLDER)
-if not os.path.exists(OUT_FOLDER):
+if not os.path.exists(os.path.join(os.getcwd(), OUT_FOLDER)):
     os.makedirs(OUT_FOLDER)
 
 
@@ -49,7 +49,7 @@ def index():
         logger.info(user_id)
         anno_path = get_anno_path(user_id)
         with open(anno_path, 'w') as f:
-            f.write("image,category,name,xMin,xMax,yMin,yMax\n")
+            f.write("image,category,name,xMin,yMin,xMax,yMax\n")
 
     if request.method == 'POST':
         if 'file' not in request.files:
@@ -79,10 +79,11 @@ def tagger():
         app.config["HEAD"] = 0
         return redirect(url_for('final'))
     user_id = session.get('_id')
-    img_dir = get_images_directory(user_id)  # app.config["IMAGES"] + f'/{user_id}'
+    # app.config["IMAGES"] + f'/{user_id}'
+    img_dir = get_images_directory(user_id)
     image = app.config["FILES"][app.config["HEAD"]]
     labels = app.config["LABELS"]
-    not_end = not(app.config["HEAD"] == len(app.config["FILES"]) - 1)
+    not_end = not (app.config["HEAD"] == len(app.config["FILES"]) - 1)
     return render_template('tagger.html', not_end=not_end, directory=img_dir, image=image, labels=labels, head=app.config["HEAD"] + 1, len=len(app.config["FILES"]))
 
 
@@ -95,12 +96,12 @@ def next():
     with open(anno_path, 'a+') as f:
         for label in app.config["LABELS"]:
             f.write(image + "," +
-            str(label["category"]) + "," +
-            label["name"] + "," +
-            str(round(float(label["xMin"]))) + "," +
-            str(round(float(label["xMax"]))) + "," +
-            str(round(float(label["yMin"]))) + "," +
-            str(round(float(label["yMax"]))) + "\n")
+                    str(label["category"]) + "," +
+                    label["name"] + "," +
+                    str(round(float(label["xMin"]))) + "," +
+                    str(round(float(label["yMin"]))) + "," +
+                    str(round(float(label["xMax"]))) + "," +
+                    str(round(float(label["yMax"]))) + "\n")
     app.config["LABELS"] = []
     return redirect(url_for('tagger'))
 
@@ -116,7 +117,8 @@ def add(id):
     xMax = request.args.get("xMax")
     yMin = request.args.get("yMin")
     yMax = request.args.get("yMax")
-    app.config["LABELS"].append({"id":id, "name":"", "category": "", "xMin":xMin, "xMax":xMax, "yMin":yMin, "yMax":yMax})
+    app.config["LABELS"].append(
+        {"id": id, "name": "", "category": "", "xMin": xMin, "xMax": xMax, "yMin": yMin, "yMax": yMax})
     return redirect(url_for('tagger'))
 
 
@@ -155,7 +157,29 @@ def download():
     user_id = session.get('_id')
     anno_path = get_anno_path(user_id)
     img_dir = get_images_directory(user_id)
-    shutil.copyfile(anno_path, f'{img_dir}/annotations_pascal_voc.csv')
+    data = {}
+    with open(anno_path, 'r') as f:
+        for line in f.readlines():
+            elems = line.strip().split(',')
+            file_name = data.get(elems[0], None)
+            if file_name is None:
+                data[elems[0]] = {'bbox': [], 'categories': [], 'names': []}
+            data[elems[0]]['categories'].append(int(elems[1]))
+            data[elems[0]]['names'].append(elems[2])
+            data[elems[0]]['bbox'].append([int(elems[i]) for i in range(3, len(elems))])
+    img_dataset = []
+    for file_name, obj in data.items():
+        img_dataset.append(
+            {
+                'file_name': file_name,
+                'objects': obj
+            }
+        )
+    with open(f'{img_dir}/metadata.jsonl', 'w') as f:
+        for item in img_dataset:
+            f.write(json.dumps(item) + "\n")
+
+    #shutil.copyfile(anno_path, f'{img_dir}/annotations_pascal_voc.csv')
     download_zip = os.path.join(img_dir, 'final')
     print(download_zip)
     shutil.make_archive('final', 'zip', img_dir)