pavel/polyfit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor tweaks

Browse Source
This commit is contained in:
Pavel Lutskov
2025-08-02 16:43:03 +02:00
parent 9225f31fca
commit 8b0c217126
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

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