Update utils.py

utils.py

@@ -8,7 +8,7 @@ from config import *
 import shutil
 from PIL import Image#上传保存遮罩和比较遮罩width height能否和章节图片匹配的时候用
 from taskMap import *
-
+from collections import Counter#计数删除少部分size不同图片
 
 def load_config():
     return get_variables()
@@ -117,6 +117,30 @@ def update_config_json(newConfig:str):
     return content
 
 
+#void
+def remove_different_size_images(chapter_path:str):
+    image_dir = chapter_path
+    # 获取目录下所有图片文件名
+    image_files = [f for f in os.listdir(image_dir) if os.path.isfile(os.path.join(image_dir, f))]
+
+    # 获取图片尺寸
+    image_sizes = []
+    for image_file in image_files:
+        with Image.open(os.path.join(image_dir, image_file)) as img:
+            image_sizes.append(img.size)
+
+    # 统计图片尺寸出现次数
+    size_counter = Counter(image_sizes)
+
+    # 找到占多数的章节图片尺寸
+    most_common_size = size_counter.most_common(1)[0][0]
+
+    # 删除其他尺寸的图片
+    for image_file, image_size in zip(image_files, image_sizes):
+        if image_size != most_common_size:
+            print("发现图片:",image_file,"大小不同于章节其他图片,认为是杂类图片,进行删除")
+            os.remove(os.path.join(image_dir, image_file))
+
 def is_asp_task_valid()->bool:
     # 启动状态检测 ->
     # 1config允许定时任务   2 mask目录下齐备 3 manga_all目录下有可用素材  4 mask的width height和素材匹配
@@ -124,7 +148,8 @@ def is_asp_task_valid()->bool:
     # 6应该避免任务扎堆进行,对ocr space造成负担
     print("判断任务条件时对config进行更新")
     configData.update()
-    
+
+
     if configData.data['allow_scheduler'] == False:
         print("当前配置不允许定时任务,停止定时任务")
         return False
@@ -132,6 +157,9 @@ def is_asp_task_valid()->bool:
     if cur_chapter == None:
         print("当前没有可用章节了,停止定时任务")
         return False
+    print("开始删除当前chapter中部分size不同的少类图片")
+    remove_different_size_images(chapter_path=cur_chapter)
+
     mask_file_path = os.path.join(configData.data['mask_dir'], '0.jpg')
     if not os.path.exists(mask_file_path):
         print("遮罩文件不存在,停止定时任务")
@@ -152,8 +180,9 @@ def is_asp_task_valid()->bool:
 
         # 现在random_chapter_image中包含了选中的随机图片文件
         if mask_image.size != random_chapter_image.size:
-            print("遮罩大小和章节随机一个图片的大小不匹配,停止定时任务")
-            return False
+            print("遮罩大小和章节随机图片的大小不匹配,继续定时任务,此处只做记录,因为前面已经删除了少类图片,且space也可以微调mask")
+            print("遮罩大小是:",mask_image.size,"随机章节图片大小是:",random_chapter_image.size)
+            return True
     else:
         print("No image files found in the directory.")
         return False
@@ -215,7 +244,7 @@ with gr.Blocks() as mangaManager:
 def process_task_list(input_task_str:str = None,baseUrl:str =None):
     if not is_asp_task_valid():
         print("任务基本条件不符合")
-        
+
         return None
     if baseUrl is None:
         print("baseUrl不存在")