From 8b0c21712633c6d17acb5cf55fa1ea29381d8013 Mon Sep 17 00:00:00 2001
From: Pavel Lutskov <pavel.lutskov@gmail.com>
Date: Sat, 2 Aug 2025 16:43:03 +0200
Subject: [PATCH] minor tweaks

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

diff --git a/main.py b/main.py
index 651136c..b3166d8 100644
--- a/main.py
+++ b/main.py
@@ -6,6 +6,7 @@ import cv2
 import numpy as np
 import torch
 
+WINSIZE = (500, 500)
 WIN = "window"
 SPACING = 50
 GREEN = (0, 255, 0)
@@ -153,12 +154,9 @@ def fit_to_line(line, *, canvas):
 
         loss = (
             0
-            #
             + point_to_point_error()
             + point_to_line_error() * 2
             + line_to_line_error()
-            #
-            # + spacing_error(fit)
         )
 
         loss.backward()
@@ -199,7 +197,7 @@ def main():
                 line.append((x, y))
                 optimization()
 
-    canvas = np.zeros((500, 500, 3), dtype=np.uint8)
+    canvas = np.zeros((*WINSIZE, 3), dtype=np.uint8)
     cv2.namedWindow(WIN)
     cv2.imshow(WIN, canvas)
     cv2.waitKey(1)