added ruff code reivew

This commit is contained in:
ykcheungab
2026-09-27 19:21:55 +08:00
parent 07da372204
commit e2ac36269c
110 changed files with 890 additions and 426 deletions
+4 -6
View File
@@ -22,7 +22,6 @@ import os
import sys
from dataclasses import dataclass, field
from pathlib import Path
from typing import Optional
import cv2
import numpy as np
@@ -35,7 +34,6 @@ for path in (REPO_ROOT, SRC_ROOT):
from orisys.tracker import Tracker # noqa: E402
DEFAULT_CONFIG = "box"
@@ -125,7 +123,7 @@ class FixedRefDIS:
preset = dis_preset_from_name(config_tracker.get("dis_preset", "medium"))
self._dis = cv2.DISOpticalFlow_create(preset)
self._dis.setFinestScale(int(config_tracker.get("dis_mode", 3)))
self._ref: Optional[np.ndarray] = None
self._ref: np.ndarray | None = None
def reset(self) -> None:
self._ref = None
@@ -148,7 +146,7 @@ def make_tracker(config_tracker: dict, *, adaptive: bool) -> Tracker:
)
def read_gray_frame(cap: cv2.VideoCapture, static_image: Optional[np.ndarray]) -> tuple[bool, Optional[np.ndarray]]:
def read_gray_frame(cap: cv2.VideoCapture, static_image: np.ndarray | None) -> tuple[bool, np.ndarray | None]:
if static_image is not None:
return True, static_image.copy()
ok, frame = cap.read()
@@ -164,11 +162,11 @@ def run_comparison(
*,
config_tracker: dict,
max_frames: int,
reset_every: Optional[int],
reset_every: int | None,
show: bool,
) -> dict[str, ModeSeries]:
path = Path(str(source)) if isinstance(source, str) else None
static_image: Optional[np.ndarray] = None
static_image: np.ndarray | None = None
if isinstance(source, str) and path is not None and path.is_file():
img = cv2.imread(str(path), cv2.IMREAD_GRAYSCALE)
if img is not None: