perf: optimize memory and performance

2026-04-17 02:09:24 +03:00
parent 16afa6308e
commit f56e5634de
1 changed files with 1 additions and 0 deletions
@@ -177,6 +177,7 @@ class VideoTAADLAA:
        sx, sy = F.conv2d(gray, net.sobel_x, padding=1), F.conv2d(gray, net.sobel_y, padding=1)
        
        # Hipotenüs hesabı ile net kenar şiddeti bulunur
+        # NaN oluşumunu önlemek için küçük epsilon eklenir
        edge = torch.sqrt(sx*sx + sy*sy + 1e-6)
        
        # Belirlenen eşik (threshold) değerine göre maske oluşturulur