MediaWiki:Gadget-LabelScan.js: Unterschied zwischen den Versionen
Admin (Diskussion | Beiträge) Keine Bearbeitungszusammenfassung Markierung: Manuelle Zurücksetzung |
Admin (Diskussion | Beiträge) Keine Bearbeitungszusammenfassung |
||
| (16 dazwischenliegende Versionen desselben Benutzers werden nicht angezeigt) | |||
| Zeile 10: | Zeile 10: | ||
modelId: 'Xenova/clip-vit-base-patch32', | modelId: 'Xenova/clip-vit-base-patch32', | ||
localModelPath: '/models', // <<— deine Modelle liegen hier | localModelPath: '/models', // <<— deine Modelle liegen hier | ||
topK: | |||
topK: 3, // <<— MAX. 3 TREFFER | |||
maxSide: 1280, // Downscale vor Auto-Crop (Performance) | maxSide: 1280, // Downscale vor Auto-Crop (Performance) | ||
// ---- Auto-Crop | // ---- Auto-Crop ---- | ||
autoCrop: true, | autoCrop: true, | ||
edgeKeepRatio: 0.10, // oberste 10% Kanten als Maske | edgeKeepRatio: 0.10, // oberste 10% Kanten als Maske | ||
| Zeile 19: | Zeile 20: | ||
cropMinRel: 0.40, // min. 40% der kleineren Bildkante | cropMinRel: 0.40, // min. 40% der kleineren Bildkante | ||
// ---- Score-Badges | // ---- Score-Badges ---- | ||
showNumericScore: false, // true = Zahlen zeigen, false = Badges | showNumericScore: false, // true = Zahlen zeigen, false = Badges | ||
confidenceBands: [0.90, 0.80], // hoch ≥0.90, mittel ≥0.80, sonst niedrig | confidenceBands: [0.90, 0.80], // hoch ≥0.90, mittel ≥0.80, sonst niedrig | ||
| Zeile 38: | Zeile 39: | ||
else{ bar.hidden=false; bar.value=Math.max(0,Math.min(1,p)); } | else{ bar.hidden=false; bar.value=Math.max(0,Math.min(1,p)); } | ||
}; | }; | ||
function resetResultsBox(msg){ | |||
const r = qs('ados-scan-results'); | |||
if (r) { | |||
r.innerHTML = `<div class="empty">${msg || 'Hier erscheinen Treffer.'}</div>`; | |||
} | |||
} | |||
function showPreview(file){ | function showPreview(file){ | ||
| Zeile 43: | Zeile 51: | ||
const prev=qs('ados-scan-preview'); | const prev=qs('ados-scan-preview'); | ||
if(prev){ | if(prev){ | ||
prev.innerHTML=`<img alt="Vorschau" style="max-width:100%;height:auto;border-radius:8px;" src="${url}">`; | prev.innerHTML=`<img alt="Vorschau" style="max-width:260px;width:100%;height:auto;border-radius:8px;display:block;margin:0 auto;" src="${url}">`; | ||
prev.setAttribute('aria-hidden','false'); | prev.setAttribute('aria-hidden','false'); | ||
} | } | ||
| Zeile 87: | Zeile 95: | ||
mod.env.allowRemoteModels = false; | mod.env.allowRemoteModels = false; | ||
mod.env.localModelPath = CFG.localModelPath; | mod.env.localModelPath = CFG.localModelPath; | ||
// WASM-Runtime (ort-wasm-simd.wasm) von transformers-CDN | // WASM-Runtime (ort-wasm-simd.wasm) von transformers-CDN | ||
| Zeile 249: | Zeile 252: | ||
function cosine(a,b){ let s=0,L=Math.min(a.length,b.length); for(let i=0;i<L;i++) s+=a[i]*b[i]; return s; } | function cosine(a,b){ let s=0,L=Math.min(a.length,b.length); for(let i=0;i<L;i++) s+=a[i]*b[i]; return s; } | ||
// vorher: slice(0, topK) hier | |||
// jetzt: ALLE sortiert zurückgeben, damit wir danach deduplizieren können | |||
function rankByCosine(q){ | function rankByCosine(q){ | ||
const s=[]; | const s=[]; | ||
| Zeile 257: | Zeile 263: | ||
} | } | ||
s.sort((a,b)=> b.score-a.score); | s.sort((a,b)=> b.score-a.score); | ||
return s | return s; | ||
} | } | ||
// --------- Score-Badges | // NEU: pro Titel nur bester Treffer | ||
function dedupeByTitle(ranked){ | |||
const bestByTitle = Object.create(null); | |||
for (let k = 0; k < ranked.length; k++) { | |||
const hit = ranked[k]; | |||
const it = INDEX[hit.i]; | |||
const rawTitle = it && it.title ? String(it.title) : ''; | |||
const key = rawTitle.trim().toLowerCase(); | |||
if (!key) continue; | |||
const prev = bestByTitle[key]; | |||
if (!prev || hit.score > prev.score) { | |||
bestByTitle[key] = hit; | |||
} | |||
} | |||
const arr = []; | |||
for (const key in bestByTitle) { | |||
if (Object.prototype.hasOwnProperty.call(bestByTitle, key)) { | |||
arr.push(bestByTitle[key]); | |||
} | |||
} | |||
arr.sort(function(a,b){ return b.score - a.score; }); | |||
return arr; | |||
} | |||
// --------- Score-Badges --------- | |||
function scoreBadge(score){ | |||
if (CFG.showNumericScore) { | if (CFG.showNumericScore) { | ||
return `<span style="font-variant-numeric:tabular-nums;color:#666">${score.toFixed(3)}</span>`; | |||
} | } | ||
const [hi, mid] = CFG.confidenceBands || [0.90, 0.80]; | const [hi, mid] = CFG.confidenceBands || [0.90, 0.80]; | ||
| Zeile 272: | Zeile 302: | ||
} | } | ||
// --------- Rendering ( | // --------- Rendering (max. 3 Treffer) --------- | ||
// Score-Badge ausgeblendet | |||
// <div>${scoreBadge(score)}</div> | |||
function renderResults(ranked){ | function renderResults(ranked){ | ||
const box=qs('ados-scan-results'); | const box=qs('ados-scan-results'); | ||
| Zeile 282: | Zeile 315: | ||
return; | return; | ||
} | } | ||
// NEU: Dedupe nach Titel, DANN auf topK begrenzen | |||
const uniqueRanked = dedupeByTitle(ranked).slice(0, CFG.topK); | |||
const makeCard = (it, score) => ` | const makeCard = (it, score) => ` | ||
| Zeile 291: | Zeile 327: | ||
<a href="${mw.util.getUrl((it.title||'').replace(/ /g,'_'))}">${mw.html.escape(it.title||'')}</a> | <a href="${mw.util.getUrl((it.title||'').replace(/ /g,'_'))}">${mw.html.escape(it.title||'')}</a> | ||
</div> | </div> | ||
<div> | <div> | ||
<a href="${mw.util.getUrl((it.title||'').replace(/ /g,'_'))}" class="mw-ui-button" style="display:inline-block;padding:.4rem .7rem;border-radius:8px;background:#2a4b8d;color:#fff;text-decoration:none;">Artikel öffnen</a> | <a href="${mw.util.getUrl((it.title||'').replace(/ /g,'_'))}" class="mw-ui-button" style="display:inline-block;padding:.4rem .7rem;border-radius:8px;background:#2a4b8d;color:#fff;text-decoration:none;">Artikel öffnen</a> | ||
| Zeile 303: | Zeile 338: | ||
grid.style.gap='12px'; | grid.style.gap='12px'; | ||
// | // max. CFG.topK (=3) Karten nach Dedupe | ||
uniqueRanked.forEach(function(hit){ | |||
const it=INDEX[i]; | const it = INDEX[hit.i]; | ||
grid.innerHTML += makeCard(it, score); | grid.innerHTML += makeCard(it, hit.score); | ||
}); | }); | ||
box.appendChild(grid); | box.appendChild(grid); | ||
| Zeile 355: | Zeile 361: | ||
if(!btnRun || !inCam || !inGal) return; | if(!btnRun || !inCam || !inGal) return; | ||
// Hilfsfunktion: wenn neues Bild gewählt → Vorschau & Ergebnisse zurücksetzen | |||
const onNewImage = (file) => { | |||
if (!file) return; | |||
showPreview(file); | |||
resetResultsBox('Hier erscheinen Treffer.'); | |||
setStatus('Bereit.'); | |||
setProgress(null); | |||
}; | |||
btnCam && btnCam.addEventListener('click', ()=> inCam.click()); | btnCam && btnCam.addEventListener('click', ()=> inCam.click()); | ||
btnGal && btnGal.addEventListener('click', ()=> inGal.click()); | btnGal && btnGal.addEventListener('click', ()=> inGal.click()); | ||
const pick = e => { const f=e.target.files?.[0]; if(f) | |||
const pick = e => { | |||
const f=e.target.files?.[0]; | |||
if(f) onNewImage(f); | |||
}; | |||
inCam.addEventListener('change', pick); | inCam.addEventListener('change', pick); | ||
inGal.addEventListener('change', pick); | inGal.addEventListener('change', pick); | ||
if(drop){ | if(drop){ | ||
drop.addEventListener('dragover', ev | drop.addEventListener('dragover', function(ev){ ev.preventDefault(); drop.classList.add('is-over'); }); | ||
drop.addEventListener('dragleave', () | drop.addEventListener('dragleave', function(){ drop.classList.remove('is-over'); }); | ||
drop.addEventListener('drop', ev | drop.addEventListener('drop', function(ev){ | ||
ev.preventDefault(); drop.classList.remove('is-over'); | ev.preventDefault(); drop.classList.remove('is-over'); | ||
const f = ev.dataTransfer | const f = ev.dataTransfer && ev.dataTransfer.files && ev.dataTransfer.files[0]; | ||
if(f){ | if(f){ | ||
const dt=new DataTransfer(); dt.items.add(f); | const dt=new DataTransfer(); dt.items.add(f); | ||
inGal.files=dt.files; | inGal.files=dt.files; | ||
onNewImage(f); | |||
} | } | ||
}); | }); | ||
} | } | ||
btnReset && btnReset.addEventListener('click', () | btnReset && btnReset.addEventListener('click', function(){ | ||
setStatus('Bereit.'); setProgress(null); | setStatus('Bereit.'); setProgress(null); | ||
const p=qs('ados-scan-preview'); if(p) p.innerHTML='<div class="note">Noch keine Vorschau.</div>'; | const p=qs('ados-scan-preview'); if(p) p.innerHTML='<div class="note">Noch keine Vorschau.</div>'; | ||
resetResultsBox('Hier erscheinen Treffer.'); | |||
inCam.value=''; inGal.value=''; | inCam.value=''; inGal.value=''; | ||
}); | }); | ||
| Zeile 388: | Zeile 408: | ||
async function onRunClick(){ | async function onRunClick(){ | ||
const btnRun = qs('ados-scan-run'); | |||
const inCam = qs('ados-scan-file-camera'); | |||
const inGal = qs('ados-scan-file-gallery'); | |||
try{ | try{ | ||
const | const file = (inCam.files && inCam.files[0]) || (inGal.files && inGal.files[0]); | ||
if(!file){ alert('Bitte zuerst ein Foto auswählen.'); return; } | if(!file){ alert('Bitte zuerst ein Foto auswählen.'); return; } | ||
if(btnRun) btnRun.disabled = true; | if(btnRun) btnRun.disabled = true; | ||
// Ergebnisse direkt leeren / „Suche läuft …“ | |||
resetResultsBox('Suche läuft …'); | |||
await loadIndex({ ui:true }); | await loadIndex({ ui:true }); | ||
| Zeile 414: | Zeile 437: | ||
err('Fehler', e); | err('Fehler', e); | ||
setStatus('Fehler bei Erkennung.'); | setStatus('Fehler bei Erkennung.'); | ||
resetResultsBox('Fehler bei der Erkennung – bitte ein anderes Foto versuchen.'); | |||
setProgress(null); | setProgress(null); | ||
} finally { | } finally { | ||
if(btnRun) btnRun.disabled = false; | if(btnRun) btnRun.disabled = false; | ||
} | } | ||
| Zeile 427: | Zeile 450: | ||
bindUI(); | bindUI(); | ||
} | } | ||
loadIndex({ ui:false }).catch(err).finally(()=>{ setStatus('Bereit.'); setProgress(null); }); | loadIndex({ ui:false }).catch(err).finally(()=>{ | ||
setStatus('Bereit.'); | |||
resetResultsBox('Hier erscheinen Treffer.'); | |||
setProgress(null); | |||
}); | |||
} | } | ||