Update app.py

app.py

@@ -26,7 +26,7 @@ os.system(
 
 ckpt = r"models/AnimeInstanceSegmentation/rtmdetl_e60.ckpt"
 
-mask_thres = 0.3
+mask_thres = 0.7
 instance_thres = 0.3
 refine_kwargs = {
     "refine_method": "refinenet_isnet"
@@ -64,8 +64,11 @@ def fn(image):
         # 把mask转换为bool类型，方便后续操作
         mask = mask.astype(np.bool_)
 
+        # 对掩码进行高斯模糊，平滑边缘
+        mask_smoothed = cv2.GaussianBlur(mask.astype(np.float32), (3, 3), 0)
+
         # 用原图中对应的区域替换白色图像中的区域，实现去除背景的效果
-        white[mask] = img2[mask]
+        white[mask_smoothed > 0.5] = img2[mask_smoothed > 0.5]
 
         # 给每个人物编号，然后用cv2.imwrite函数来保存图片到文件夹中
         filename = f"person_{ii+1}.png"