Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| b143c6607a |
+12
-85
@@ -110,62 +110,6 @@ if HAS_NUMBA:
|
|||||||
acc[y, x] *= inv
|
acc[y, x] *= inv
|
||||||
return acc
|
return acc
|
||||||
|
|
||||||
@nb.njit(cache=True, parallel=True, fastmath=True, boundscheck=False)
|
|
||||||
def _jit_score_bitmap_greedy(
|
|
||||||
spread: np.ndarray,
|
|
||||||
dx: np.ndarray, dy: np.ndarray, bins: np.ndarray,
|
|
||||||
bit_active: np.uint8,
|
|
||||||
min_score: nb.float32,
|
|
||||||
greediness: nb.float32,
|
|
||||||
) -> np.ndarray:
|
|
||||||
"""Score bitmap con early-exit greedy (no rescore background).
|
|
||||||
|
|
||||||
Per ogni pixel iteriamo le N feature; abortiamo non appena diventa
|
|
||||||
impossibile raggiungere `min_required` count anche aggiungendo
|
|
||||||
tutte le feature rimanenti. min_required = greediness * min_score * N.
|
|
||||||
|
|
||||||
greediness=0 → nessun early-exit (equivalente a kernel base).
|
|
||||||
greediness=1 → exit non appena hits + remaining < min_score * N.
|
|
||||||
Tipico: 0.7-0.9 → 2-4x speed-up senza perdere match.
|
|
||||||
"""
|
|
||||||
H, W = spread.shape
|
|
||||||
N = dx.shape[0]
|
|
||||||
acc = np.zeros((H, W), dtype=np.float32)
|
|
||||||
if N == 0:
|
|
||||||
return acc
|
|
||||||
min_req = greediness * min_score * N
|
|
||||||
inv_N = nb.float32(1.0 / N)
|
|
||||||
for y in nb.prange(H):
|
|
||||||
for x in range(W):
|
|
||||||
hits = 0
|
|
||||||
for i in range(N):
|
|
||||||
b = bins[i]
|
|
||||||
mask = np.uint8(1) << b
|
|
||||||
if (bit_active & mask) == 0:
|
|
||||||
# Nessun chance per questa feature
|
|
||||||
if hits + (N - i - 1) < min_req:
|
|
||||||
break
|
|
||||||
continue
|
|
||||||
ddy = dy[i]
|
|
||||||
yy = y + ddy
|
|
||||||
if yy < 0 or yy >= H:
|
|
||||||
if hits + (N - i - 1) < min_req:
|
|
||||||
break
|
|
||||||
continue
|
|
||||||
ddx = dx[i]
|
|
||||||
xx = x + ddx
|
|
||||||
if xx < 0 or xx >= W:
|
|
||||||
if hits + (N - i - 1) < min_req:
|
|
||||||
break
|
|
||||||
continue
|
|
||||||
if spread[yy, xx] & mask:
|
|
||||||
hits += 1
|
|
||||||
else:
|
|
||||||
if hits + (N - i - 1) < min_req:
|
|
||||||
break
|
|
||||||
acc[y, x] = nb.float32(hits) * inv_N
|
|
||||||
return acc
|
|
||||||
|
|
||||||
@nb.njit(cache=True, parallel=True, fastmath=True, boundscheck=False)
|
@nb.njit(cache=True, parallel=True, fastmath=True, boundscheck=False)
|
||||||
def _jit_score_bitmap_rescored(
|
def _jit_score_bitmap_rescored(
|
||||||
spread: np.ndarray, # uint8 (H, W)
|
spread: np.ndarray, # uint8 (H, W)
|
||||||
@@ -241,10 +185,6 @@ if HAS_NUMBA:
|
|||||||
_jit_score_bitmap(spread, dx, dy, b, np.uint8(0xFF))
|
_jit_score_bitmap(spread, dx, dy, b, np.uint8(0xFF))
|
||||||
bg = np.zeros((32, 32), dtype=np.float32)
|
bg = np.zeros((32, 32), dtype=np.float32)
|
||||||
_jit_score_bitmap_rescored(spread, dx, dy, b, np.uint8(0xFF), bg)
|
_jit_score_bitmap_rescored(spread, dx, dy, b, np.uint8(0xFF), bg)
|
||||||
_jit_score_bitmap_greedy(
|
|
||||||
spread, dx, dy, b, np.uint8(0xFF),
|
|
||||||
np.float32(0.5), np.float32(0.8),
|
|
||||||
)
|
|
||||||
_jit_popcount_density(spread)
|
_jit_popcount_density(spread)
|
||||||
|
|
||||||
else: # pragma: no cover
|
else: # pragma: no cover
|
||||||
@@ -258,9 +198,6 @@ else: # pragma: no cover
|
|||||||
def _jit_score_bitmap_rescored(spread, dx, dy, bins, bit_active, bg):
|
def _jit_score_bitmap_rescored(spread, dx, dy, bins, bit_active, bg):
|
||||||
raise RuntimeError("numba non disponibile")
|
raise RuntimeError("numba non disponibile")
|
||||||
|
|
||||||
def _jit_score_bitmap_greedy(spread, dx, dy, bins, bit_active, min_score, greediness):
|
|
||||||
raise RuntimeError("numba non disponibile")
|
|
||||||
|
|
||||||
def _jit_popcount_density(spread):
|
def _jit_popcount_density(spread):
|
||||||
raise RuntimeError("numba non disponibile")
|
raise RuntimeError("numba non disponibile")
|
||||||
|
|
||||||
@@ -309,32 +246,22 @@ def score_bitmap_rescored(
|
|||||||
return np.maximum(0.0, out).astype(np.float32)
|
return np.maximum(0.0, out).astype(np.float32)
|
||||||
|
|
||||||
|
|
||||||
def score_bitmap_greedy(
|
_HAS_NP_BITCOUNT = hasattr(np, "bitwise_count")
|
||||||
spread: np.ndarray, dx: np.ndarray, dy: np.ndarray, bins: np.ndarray,
|
|
||||||
bit_active: int, min_score: float, greediness: float,
|
|
||||||
) -> np.ndarray:
|
|
||||||
"""Score bitmap con early-exit greedy. Per coarse-pass aggressivo.
|
|
||||||
|
|
||||||
Non applica rescore background: usare quando la scena ha basso clutter
|
|
||||||
o quando si vuole mass-prune varianti via top-level rapidamente.
|
|
||||||
"""
|
|
||||||
if HAS_NUMBA and len(dx) > 0:
|
|
||||||
return _jit_score_bitmap_greedy(
|
|
||||||
np.ascontiguousarray(spread, dtype=np.uint8),
|
|
||||||
np.ascontiguousarray(dx, dtype=np.int32),
|
|
||||||
np.ascontiguousarray(dy, dtype=np.int32),
|
|
||||||
np.ascontiguousarray(bins, dtype=np.int8),
|
|
||||||
np.uint8(bit_active),
|
|
||||||
np.float32(min_score), np.float32(greediness),
|
|
||||||
)
|
|
||||||
# Fallback: kernel base senza early-exit
|
|
||||||
return score_bitmap(spread, dx, dy, bins, bit_active)
|
|
||||||
|
|
||||||
|
|
||||||
def popcount_density(spread: np.ndarray) -> np.ndarray:
|
def popcount_density(spread: np.ndarray) -> np.ndarray:
|
||||||
|
"""Conta bit set per pixel.
|
||||||
|
|
||||||
|
Order:
|
||||||
|
1) Numba JIT parallel (preferito: piu veloce su 1080p, 0.5ms vs 1.6ms)
|
||||||
|
2) numpy.bitwise_count (NumPy 2.0+, SIMD ma single-thread)
|
||||||
|
3) Fallback numpy bit-shift puro
|
||||||
|
"""
|
||||||
|
spread_c = np.ascontiguousarray(spread, dtype=np.uint8)
|
||||||
if HAS_NUMBA:
|
if HAS_NUMBA:
|
||||||
return _jit_popcount_density(np.ascontiguousarray(spread, dtype=np.uint8))
|
return _jit_popcount_density(spread_c)
|
||||||
# Fallback
|
if _HAS_NP_BITCOUNT:
|
||||||
|
return np.bitwise_count(spread_c).astype(np.float32, copy=False)
|
||||||
H, W = spread.shape
|
H, W = spread.shape
|
||||||
out = np.zeros((H, W), dtype=np.float32)
|
out = np.zeros((H, W), dtype=np.float32)
|
||||||
for b in range(8):
|
for b in range(8):
|
||||||
|
|||||||
+1
-13
@@ -40,7 +40,6 @@ from pm2d._jit_kernels import (
|
|||||||
score_by_shift as _jit_score_by_shift,
|
score_by_shift as _jit_score_by_shift,
|
||||||
score_bitmap as _jit_score_bitmap,
|
score_bitmap as _jit_score_bitmap,
|
||||||
score_bitmap_rescored as _jit_score_bitmap_rescored,
|
score_bitmap_rescored as _jit_score_bitmap_rescored,
|
||||||
score_bitmap_greedy as _jit_score_bitmap_greedy,
|
|
||||||
popcount_density as _jit_popcount,
|
popcount_density as _jit_popcount,
|
||||||
HAS_NUMBA,
|
HAS_NUMBA,
|
||||||
)
|
)
|
||||||
@@ -575,7 +574,6 @@ class LineShapeMatcher:
|
|||||||
verify_threshold: float = 0.4,
|
verify_threshold: float = 0.4,
|
||||||
coarse_angle_factor: int = 2,
|
coarse_angle_factor: int = 2,
|
||||||
scale_penalty: float = 0.0,
|
scale_penalty: float = 0.0,
|
||||||
greediness: float = 0.0,
|
|
||||||
) -> list[Match]:
|
) -> list[Match]:
|
||||||
"""
|
"""
|
||||||
scale_penalty: se > 0, riduce lo score per match a scala diversa da 1.0:
|
scale_penalty: se > 0, riduce lo score per match a scala diversa da 1.0:
|
||||||
@@ -647,20 +645,10 @@ class LineShapeMatcher:
|
|||||||
end = min(n, i + half + 1)
|
end = min(n, i + half + 1)
|
||||||
neighbor_map[vi_c] = vi_sorted[start:end]
|
neighbor_map[vi_c] = vi_sorted[start:end]
|
||||||
|
|
||||||
# Pruning varianti via top-level (parallelizzato) - solo coarse.
|
# Pruning varianti via top-level (parallelizzato) - solo coarse
|
||||||
# greediness > 0: usa kernel greedy con early-exit (no rescore bg)
|
|
||||||
# per il pruning. ~2-4x speed-up sul top con greediness=0.8.
|
|
||||||
use_greedy_top = greediness > 0.0
|
|
||||||
|
|
||||||
def _top_score(vi: int) -> tuple[int, float]:
|
def _top_score(vi: int) -> tuple[int, float]:
|
||||||
var = self.variants[vi]
|
var = self.variants[vi]
|
||||||
lvl = var.levels[min(top, len(var.levels) - 1)]
|
lvl = var.levels[min(top, len(var.levels) - 1)]
|
||||||
if use_greedy_top:
|
|
||||||
score = _jit_score_bitmap_greedy(
|
|
||||||
spread_top, lvl.dx, lvl.dy, lvl.bin, bit_active_top,
|
|
||||||
top_thresh, greediness,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
score = _jit_score_bitmap_rescored(
|
score = _jit_score_bitmap_rescored(
|
||||||
spread_top, lvl.dx, lvl.dy, lvl.bin, bit_active_top,
|
spread_top, lvl.dx, lvl.dy, lvl.bin, bit_active_top,
|
||||||
bg_cache_top[var.scale],
|
bg_cache_top[var.scale],
|
||||||
|
|||||||
Reference in New Issue
Block a user