From c62bde0efac2b9466252ec6c9a3ea4e665a6c506 Mon Sep 17 00:00:00 2001
From: wrz1-zzzzz <914921336@qq.com>
Date: Thu, 21 Nov 2024 12:50:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8F=82=E6=95=B0block=5Fsiz?=
 =?UTF-8?q?e=5Fx,=20block=5Fsize=5Fy=20=E7=94=A8=E4=BA=8E=E5=92=8C?=
 =?UTF-8?q?=E5=96=B7=E9=98=80=E5=AF=B9=E5=BA=94,=E6=AF=8F=E8=A1=8C?=
 =?UTF-8?q?=E5=83=8F=E7=B4=A0=E6=95=B0=E9=87=8F=E5=92=8C=E6=AF=8F=E5=88=97?=
 =?UTF-8?q?=E5=AF=B9=E5=BA=94=E9=AB=98=E5=BA=A6=EF=BC=88=E8=A1=8C=E9=A2=91?=
 =?UTF-8?q?=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DPL/yolov5/mask.py  | 4 +++-
 DPL/yolov5/model.py | 6 +++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/DPL/yolov5/mask.py b/DPL/yolov5/mask.py
index b885e0d..97a10a6 100644
--- a/DPL/yolov5/mask.py
+++ b/DPL/yolov5/mask.py
@@ -4,7 +4,9 @@ from model import Model
 model = Model(
     image_folder="",
     label_folder="",
-    output_folder=""
+    output_folder="",
+    block_size_x=24,
+    block_size_y=24
 )
 
 # 读取某张图片对应的掩膜
diff --git a/DPL/yolov5/model.py b/DPL/yolov5/model.py
index e58a8b3..b96ac4a 100644
--- a/DPL/yolov5/model.py
+++ b/DPL/yolov5/model.py
@@ -4,7 +4,7 @@ import numpy as np
 import time
 
 class Model:
-    def __init__(self, image_folder, label_folder, output_folder, width_blocks=24, height_blocks=24):
+    def __init__(self, image_folder, label_folder, output_folder, block_size_x=24, block_size_y=24):
         """
         初始化 Model 类，并自动生成并保存掩膜
         :param image_folder: 输入图像文件夹路径
@@ -16,8 +16,8 @@ class Model:
         self.image_folder = image_folder
         self.label_folder = label_folder
         self.output_folder = output_folder
-        self.width_blocks = width_blocks
-        self.height_blocks = height_blocks
+        self.width_blocks = block_size_x
+        self.height_blocks = block_size_y
 
         # 确保输出文件夹存在
         if not os.path.exists(self.output_folder):