333

labelimg.py

@@ -0,0 +1,94 @@
+import os
+from PIL import Image
+import imageio
+
+COMIC_DIR = "comic/Ch11A"
+
+
+def get_last_frame_gif(path):
+    gif = Image.open(path)
+    try:
+        while True:
+            gif.seek(gif.tell() + 1)
+    except EOFError:
+        return gif.convert("RGB")
+
+
+def load_image(path):
+    if path.lower().endswith(".gif"):
+        return get_last_frame_gif(path)
+    return Image.open(path).convert("RGB")
+
+
+def get_color(img, x, y):
+    return img.getpixel((x, y))
+
+
+def is_color(c, hex_code):
+    rgb = tuple(int(hex_code[i : i + 2], 16) for i in (1, 3, 5))
+    return c == rgb
+
+
+def check_diag(img, x1, y1, x2, y2, color_hex):
+    return is_color(get_color(img, x1, y1), color_hex) and is_color(
+        get_color(img, x2, y2), color_hex
+    )
+
+
+def is_battle_type(img):
+    pts = [(32, 473), (185, 432), (454, 473), (609, 432)]
+    return all(
+        is_color(get_color(img, x, y), "#ff7f27")
+        or is_color(get_color(img, x, y), "#ffff00")
+        for x, y in pts
+    )
+
+
+def has_textbox(img, base_y_offset):
+    return (
+        check_diag(img, 32, base_y_offset, 37, base_y_offset + 5, "#ffffff")
+        and check_diag(
+            img, 604, base_y_offset + 146, 609, base_y_offset + 151, "#ffffff"
+        )
+        and check_diag(img, 38, base_y_offset + 6, 42, base_y_offset + 10, "#000000")
+    )
+
+
+def check_avatar_absent(img, y1, y2):
+    def line_black(y):
+        for x in range(60, 145):
+            if get_color(img, x, y) != (0, 0, 0):
+                return False
+        return True
+
+    return line_black(y1) and line_black(y2)
+
+
+def process_image(path):
+    img = load_image(path)
+    result = {"file": os.path.basename(path)}
+
+    if is_battle_type(img):
+        result["type"] = "战斗类"
+        has_txt = has_textbox(img, 250)
+        result["战斗主文本框"] = has_txt
+    else:
+        result["type"] = "非战斗类"
+        result["下方文本框"] = has_textbox(img, 320)
+        result["上方文本框"] = has_textbox(img, 10)
+        result["下方无头像"] = check_avatar_absent(img, 390, 430)
+        result["上方无头像"] = check_avatar_absent(img, 60, 100)
+
+    return result
+
+
+def main():
+    for file in os.listdir(COMIC_DIR):
+        if file.lower().endswith((".png", ".gif")):
+            path = os.path.join(COMIC_DIR, file)
+            result = process_image(path)
+            print(result)
+
+
+if __name__ == "__main__":
+    main()

trackpoint.md

@@ -0,0 +1,46 @@
+# tranckpoint.md
+
+记录判定漫画类型和人物特征贴图需要跟踪的点位和对应颜色。
+
+## 战斗类
+识别四个特征点位：
+32,473
+185,432
+454,473
+609,432
+如果四个点位均为#ff7f27或者#ffff00，则为战斗类
+
+### 战斗主文本框：
+### 下方文本框：
+- 32,250 -- 37,255对角线 #fff
+- 604,396 -- 609,401对角线 #fff
+- 38,256 -- 42,260对角线 #000
+满足以上三条（下方文本框向上偏移70px），则说明有文本框
+
+## 非战斗类
+### 下方文本框：
+- 32,320 -- 37,325对角线 #fff
+- 604,466 -- 609,471对角线 #fff
+- 38,326 -- 42,330对角线 #000
+满足以上三条
+
+框内黑色区域：38,326 -- 639,465  大小 601*139
+
+是否存在人物贴图：
+扫描 60,390 -- 144,390
+扫描 60,430 -- 144,430
+两条线，如果均为全黑色，则说明无头像贴图
+
+
+### 上方文本框：
+- 32,10 -- 37,15对角线 #fff
+- 604,156 -- 609,161对角线 #fff
+- 38,16 -- 42,20对角线 #000
+满足以上三条（下方文本框向上偏移310px）
+
+框内黑色区域：38,16 -- 639,155  大小 601*139
+
+是否存在人物贴图：
+扫描 60,60 -- 144,60
+扫描 60,100 -- 144,100
+两条线，如果均为全黑色，则说明无头像贴图