add object category

app.py

@@ -2,6 +2,7 @@ import os
 import shutil
 import uuid
 import logging
+import json
 
 from flask import Flask, redirect, url_for, request, flash, session, after_this_request
 from flask import render_template
@@ -19,6 +20,7 @@ app.config["SECRET_KEY"] = '#thaistartsfirst!'
 app.config["IMAGES"] = UPLOAD_FOLDER
 app.config["OUT"] = OUT_FOLDER
 app.config["LABELS"] = []
+app.config["CATEGORIES"] = {} # []
 app.config["HEAD"] = 0
 app.config["SESSION_PERMANENT"] = False
 
@@ -47,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,id,name,xMin,xMax,yMin,yMax\n")
+            f.write("image,category,name,xMin,xMax,yMin,yMax\n")
 
     if request.method == 'POST':
         if 'file' not in request.files:
@@ -93,7 +95,7 @@ def next():
     with open(anno_path, 'a+') as f:
         for label in app.config["LABELS"]:
             f.write(image + "," +
-            label["id"] + "," +
+            str(label["category"]) + "," +
             label["name"] + "," +
             str(round(float(label["xMin"]))) + "," +
             str(round(float(label["xMax"]))) + "," +
@@ -114,7 +116,7 @@ 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":"", "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'))
 
 
@@ -129,8 +131,14 @@ def remove(id):
 
 @app.route('/label/<id>')
 def label(id):
-    name = request.args.get("name")
+    name = request.args.get("name").lower()
     app.config["LABELS"][int(id) - 1]["name"] = name
+    category = app.config["CATEGORIES"].get(name, None)
+    if category is None:
+        # add to category
+        category = len(app.config["CATEGORIES"])
+        app.config["CATEGORIES"][name] = category
+    app.config["LABELS"][int(id) - 1]["category"] = category
     return redirect(url_for('tagger'))
 
 
@@ -171,4 +179,4 @@ def download():
 
 
 if __name__ == "__main__":
-    app.run(debug="True")
+    app.run(debug="True")