From efac01dacaca8096b22c33c4d566b9524ca4f1b6 Mon Sep 17 00:00:00 2001
From: GF <149839341+gorillaframeai@users.noreply.github.com>
Date: Thu, 20 Feb 2025 06:57:39 +0300
Subject: [PATCH] Update gfrbmg2.py

---
 gfrbmg2.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gfrbmg2.py b/gfrbmg2.py
index 0e95641..249477f 100644
--- a/gfrbmg2.py
+++ b/gfrbmg2.py
@@ -10,7 +10,7 @@ from torchvision.transforms.functional import normalize
 import numpy as np
 import cv2
 import subprocess
-import torch.nn.functional as F  # Добавьте эту строку
+import torch.nn.functional as F
 
 script_directory = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(script_directory)
@@ -212,7 +212,7 @@ class GFrbmg2:
                     result = self.model(im_tensor)[-1].sigmoid().cpu()
 
                 result = result[0].squeeze()
-                result = F.interpolate(result.unsqueeze(0).unsqueeze(0), size=(h, w), mode='bilinear').squeeze()
+                result = F.interpolate(result.unsqueeze(0).unsqueeze(0), size=(h, w), mode='bilinear', align_corners=True).squeeze()
 
                 mask_pil = tensor2pil(result)
                 # Чистка маски с учетом всех параметров