add json

app.py

@@ -1,8 +1,7 @@
 import os
-import sys
-import time
 import glob
-import yaml
+import time
+import json
 import random
 
 import numpy as np
@@ -28,10 +27,10 @@ def resize_height(img, height, nearest=False):
     return img.resize((new_width, height), Image.NEAREST if nearest else Image.BICUBIC), new_width
 
 
-# load config yaml file
+# load config json file
 def load_config(path):
     with open(path, "r") as f:
-        config = yaml.load(f, Loader=yaml.FullLoader)
+        config = json.load(f)
     return config
 
 
@@ -92,14 +91,16 @@ def get_mask(cfg):
 
     mask, _ = resize_height(mask, 512)
     mask_wo_border = mask.copy()
-    mask_wo_border = mask_wo_border.filter(ImageFilter.MaxFilter(3)).filter(ImageFilter.GaussianBlur(1.5))
+    mask_wo_border = mask_wo_border.filter(ImageFilter.MaxFilter(3))
     mask = mask.filter(ImageFilter.MaxFilter(5)).filter(ImageFilter.GaussianBlur(1.5))
 
     ord_, _ = resize_height(ord_, 512, nearest=True)
 
-    plate_bg = get_background(name).resize(mask.size)
+    plate_bg = get_background(name)
+    mask_wo_border = mask_wo_border.resize(plate_bg.size)
+    
     # make the mask as alpha channel
-    bg = Image.new("RGBA", mask.size, (0, 0, 0, 0))
+    bg = Image.new("RGBA", plate_bg.size, (0, 0, 0, 0))
     bg.paste(plate_bg, (0, 0))
     # hex to rgb
     plateTextColor = tuple(int(plateTextColor[i:i + 2], 16) for i in (0, 2, 4))
@@ -114,7 +115,7 @@ plates = [f for f in os.listdir(config_path) if os.path.isdir(os.path.join(confi
 
 
 def generate(name):
-    cfg = load_config(os.path.join(config_path, name, "config.yaml"))
+    cfg = load_config(os.path.join(config_path, name, "config.json"))
     text, bg, mask, ord_ = get_mask(cfg)
     return text, bg, mask, ord_
 
@@ -126,10 +127,11 @@ demo = gr.Interface(
     ],
     [
         gr.Text(label="Plate Text"),
-        gr.Image(label="Background"),
-        gr.Image(label="Mask"),
-        gr.Image(label="Ord"),
-    ]
+        gr.Image(label="Background", show_label=False, format="png"),
+        gr.Image(label="Mask", show_label=False, format="png"),
+        gr.Image(label="Ord", show_label=False, format="png"),
+    ],
+    allow_flagging=False,
 )
 
 

config/cn_car/config.json

@@ -0,0 +1,46 @@
+{
+    "type": 4,
+    "name": "cn_car",
+    "plateNums": [
+        "0",
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9"
+    ],
+    "plateChars": [
+        "A",
+        "B",
+        "C",
+        "D",
+        "E",
+        "F",
+        "G",
+        "H",
+        "J",
+        "K",
+        "L",
+        "M",
+        "N",
+        "P",
+        "Q",
+        "R",
+        "S",
+        "T",
+        "U",
+        "V",
+        "W",
+        "X",
+        "Y"
+    ],
+    "plateCities": "京津冀晋蒙辽吉黑沪苏浙皖闽赣鲁豫鄂湘粤桂琼渝川贵云藏陕甘青宁新",
+    "plateTextColor": "#FFFFFF",
+    "plateFormat": [
+        "@A·#####"
+    ]
+}

config/cn_car/config.yaml

@@ -1,4 +1,4 @@
-type: 0
+type: 4
 name: "cn_car"
 plateNums: ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]
 plateChars: ["A", "B", "C", "D", "E", "F", "G", "H", "J", "K", "L", "M", "N", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y"]

config/cn_truck/config.json

@@ -0,0 +1,13 @@
+{
+    "type": 3,
+    "name": "cn_truck",
+    "multiLine": true,
+    "plateNums": "1234567890",
+    "plateChars": "ABCDEFGHJKLMNPQRSTUVWXYZ",
+    "plateCities": "京津冀晋蒙辽吉黑沪苏浙皖闽赣鲁豫鄂湘粤桂琼渝川贵云藏陕甘青宁新",
+    "plateTextColor": "#111111",
+    "plateFormat": [
+        "@·A",
+        "#0000"
+    ]
+}

config/cn_truck/config.yaml

@@ -1,9 +1,10 @@
-type: 0
+type: 3
 name: "cn_truck"
 multiLine: true
 plateNums: "1234567890"
 plateChars: "ABCDEFGHJKLMNPQRSTUVWXYZ"
 plateCities: "京津冀晋蒙辽吉黑沪苏浙皖闽赣鲁豫鄂湘粤桂琼渝川贵云藏陕甘青宁新"
+plateTextColor: "#111111"
 # format is a string including #, A, 0, 1 and @
 # # representing any number and letter
 # A representing any letter

config/de_car/config.json

@@ -0,0 +1,48 @@
+{
+    "type": 1,
+    "name": "de_car",
+    "plateNums": [
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9",
+        "0"
+    ],
+    "plateChars": [
+        "A",
+        "B",
+        "C",
+        "D",
+        "E",
+        "F",
+        "G",
+        "H",
+        "I",
+        "J",
+        "K",
+        "L",
+        "M",
+        "N",
+        "O",
+        "P",
+        "Q",
+        "R",
+        "S",
+        "T",
+        "U",
+        "V",
+        "W",
+        "X",
+        "Y",
+        "Z"
+    ],
+    "plateCities": "",
+    "plateFormat": [
+        "AAA0000"
+    ]
+}

config/de_car/config.yaml

@@ -1,4 +1,4 @@
-type: 0
+type: 1
 name: "de_car"
 plateNums: ["1", "2", "3", "4", "5", "6", "7", "8", "9", "0"]
 plateChars: ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"]

config/jp_car/config.json

@@ -0,0 +1,57 @@
+{
+    "type": 0,
+    "name": "jp_car",
+    "plateNums": [
+        "0",
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9"
+    ],
+    "plateChars": [
+        "は",
+        "ほ",
+        "ね",
+        "せ",
+        "さ",
+        "ひ",
+        "や",
+        "ろ",
+        "ま",
+        "ぬ"
+    ],
+    "plateCities": [
+        "長野",
+        "土浦",
+        "神戸",
+        "福岡",
+        "広島",
+        "大阪",
+        "多摩",
+        "品川",
+        "札幌",
+        "沖縄",
+        "東京",
+        "豊田",
+        "横浜",
+        "足立",
+        "静岡",
+        "鈴鹿",
+        "滋賀",
+        "三重",
+        "福山",
+        "愛媛",
+        "千葉",
+        "青森",
+        "群馬"
+    ],
+    "plateFormat": [
+        "@000",
+        "A00-00"
+    ]
+}

config/kr_car/config.json

@@ -0,0 +1,58 @@
+{
+    "type": 1,
+    "name": "kr_car",
+    "plateNums": [
+        "0",
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9"
+    ],
+    "plateChars": [
+        "가",
+        "나",
+        "다",
+        "라",
+        "마",
+        "거",
+        "너",
+        "더",
+        "러",
+        "머",
+        "버",
+        "서",
+        "어",
+        "저",
+        "고",
+        "노",
+        "도",
+        "로",
+        "모",
+        "보",
+        "소",
+        "오",
+        "조",
+        "구",
+        "누",
+        "두",
+        "루",
+        "무",
+        "부",
+        "수",
+        "우",
+        "주",
+        "하",
+        "허",
+        "호"
+    ],
+    "plateCities": "",
+    "plateTextColor": "#111111",
+    "plateFormat": [
+        "000A0000"
+    ]
+}

config/kr_car/config.yaml

@@ -1,8 +1,9 @@
-type: 0
+type: 1
 name: "kr_car"
 plateNums: ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]
 plateChars: ["가", "나", "다", "라", "마", "거", "너", "더", "러", "머", "버", "서", "어", "저", "고", "노", "도", "로", "모", "보", "소", "오", "조", "구", "누", "두", "루", "무", "부", "수", "우", "주", "하", "허", "호",]
 plateCities: ""
+plateTextColor: "#111111"
 # format is a string including #, A, 0, and @
 # # representing any number and letter
 # A representing any letter

config/kr_taxi/config.json

@@ -0,0 +1,44 @@
+{
+    "type": 1,
+    "name": "kr_taxi",
+    "plateNums": [
+        "0",
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9"
+    ],
+    "plateChars": [
+        "바",
+        "사",
+        "아",
+        "자"
+    ],
+    "plateCities": [
+        "서울",
+        "부산",
+        "대구",
+        "인천",
+        "광주",
+        "대전",
+        "울산",
+        "경기",
+        "강원",
+        "충북",
+        "충남",
+        "전북",
+        "전남",
+        "경북",
+        "경남",
+        "제주"
+    ],
+    "plateTextColor": "#111111",
+    "plateFormat": [
+        "@00A0000"
+    ]
+}

config/kr_taxi/config.yaml

@@ -1,8 +1,9 @@
-type: 0
+type: 1
 name: "kr_taxi"
 plateNums: ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]
 plateChars: ["바", "사", "아", "자"]
 plateCities: ["서울", "부산", "대구", "인천", "광주", "대전", "울산", "경기", "강원", "충북", "충남", "전북", "전남", "경북", "경남", "제주"]
+plateTextColor: "#111111"
 # format is a string including #, A, 0, and @
 # # representing any number and letter
 # A representing any letter

config/kr_truck/config.json

@@ -0,0 +1,45 @@
+{
+    "type": 0,
+    "name": "kr_truck",
+    "plateNums": [
+        "0",
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9"
+    ],
+    "plateChars": [
+        "바",
+        "사",
+        "아",
+        "자"
+    ],
+    "plateCities": [
+        "서울",
+        "부산",
+        "대구",
+        "인천",
+        "광주",
+        "대전",
+        "울산",
+        "경기",
+        "강원",
+        "충북",
+        "충남",
+        "전북",
+        "전남",
+        "경북",
+        "경남",
+        "제주"
+    ],
+    "plateTextColor": "#FFFFFF",
+    "plateFormat": [
+        "@00",
+        "A0000"
+    ]
+}

config/us_cali/config.json

@@ -0,0 +1,49 @@
+{
+    "type": 0,
+    "name": "us_cali",
+    "plateNums": [
+        "1",
+        "2",
+        "3",
+        "4",
+        "5",
+        "6",
+        "7",
+        "8",
+        "9",
+        "0"
+    ],
+    "plateChars": [
+        "A",
+        "B",
+        "C",
+        "D",
+        "E",
+        "F",
+        "G",
+        "H",
+        "I",
+        "J",
+        "K",
+        "L",
+        "M",
+        "N",
+        "O",
+        "P",
+        "Q",
+        "R",
+        "S",
+        "T",
+        "U",
+        "V",
+        "W",
+        "X",
+        "Y",
+        "Z"
+    ],
+    "plateCities": "",
+    "plateTextColor": "#0E1947",
+    "plateFormat": [
+        "0AAA000"
+    ]
+}

yaml2json.py

@@ -0,0 +1,24 @@
+import json
+import yaml
+import glob
+
+# convert yaml to json
+def yaml_to_json(yaml_file):
+    with open(yaml_file) as f:
+        data = yaml.safe_load(f)
+    # return data
+    basename = yaml_file.split(".")[0]
+    # save json data to a file
+    with open(f"{basename}.json", "w") as f:
+        # utf-8 encoding
+        f.write(json.dumps(data, indent=4, ensure_ascii=False))
+
+    return data
+
+
+if __name__ == "__main__":
+    # yaml_file = "test.yaml"
+    # yaml_to_json(yaml_file)
+
+    for f in glob.glob("config/*/*.yaml"):
+        yaml_to_json(f)