martin/object-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Using QPolygonF instead of drawLine

Browse Source
This commit is contained in:
Martin Laasmaa
2025-12-11 17:14:07 +02:00
parent 9a221acb63
commit b3c5a51dbb
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/gui/widgets/annotation_canvas_widget.py
View File

@@ -16,8 +16,9 @@ from PySide6.QtGui import (
QKeyEvent, QKeyEvent,
QMouseEvent, QMouseEvent,
QPaintEvent, QPaintEvent,
QPolygonF,
) )
from PySide6.QtCore import Qt, QEvent, Signal, QPoint, QRect from PySide6.QtCore import Qt, QEvent, Signal, QPoint, QPointF, QRect
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
from src.utils.image import Image, ImageLoadError from src.utils.image import Image, ImageLoadError
@@ -496,8 +497,10 @@ class AnnotationCanvasWidget(QWidget):
) )
painter.setPen(pen) painter.setPen(pen)
for (x1, y1), (x2, y2) in zip(polyline[:-1], polyline[1:]): # Use QPolygonF for efficient polygon rendering (single call vs N-1 calls)
painter.drawLine(int(x1), int(y1), int(x2), int(y2)) # drawPolygon() automatically closes the shape, ensuring proper visual closure
polygon = QPolygonF([QPointF(x, y) for x, y in polyline])
painter.drawPolygon(polygon)
# Draw bounding boxes (dashed) if enabled # Draw bounding boxes (dashed) if enabled
if self.show_bboxes and self.original_pixmap is not None and self.bboxes: if self.show_bboxes and self.original_pixmap is not None and self.bboxes: