Update app.py
Browse files
app.py
CHANGED
@@ -172,7 +172,16 @@ def submit_function(
|
|
172 |
print("Không có layers trong person_image.")
|
173 |
|
174 |
|
175 |
-
mask = Image.open(mask).convert("L") # Chuyển mặt nạ thành ảnh thang độ xám
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
176 |
if len(np.unique(np.array(mask))) == 1: # Nếu mặt nạ chỉ chứa một giá trị (ví dụ: toàn đen hoặc toàn trắng), thì không sử dụng mặt nạ (mask = None).
|
177 |
mask = None
|
178 |
else:
|
|
|
172 |
print("Không có layers trong person_image.")
|
173 |
|
174 |
|
175 |
+
#mask = Image.open(mask).convert("L") # Chuyển mặt nạ thành ảnh thang độ xám
|
176 |
+
if mask is None:
|
177 |
+
raise ValueError("Tham số 'mask' bị rỗng.")
|
178 |
+
elif isinstance(mask, (str, bytes)) or hasattr(mask, "read"):
|
179 |
+
mask = Image.open(mask).convert("L")
|
180 |
+
else:
|
181 |
+
raise ValueError(f"Kiểu dữ liệu '{type(mask)}' của 'mask' không được hỗ trợ.")
|
182 |
+
print("Loại của mask:", type(mask))
|
183 |
+
print("Giá trị của mask:", mask)
|
184 |
+
|
185 |
if len(np.unique(np.array(mask))) == 1: # Nếu mặt nạ chỉ chứa một giá trị (ví dụ: toàn đen hoặc toàn trắng), thì không sử dụng mặt nạ (mask = None).
|
186 |
mask = None
|
187 |
else:
|