HyperAIHyperAI

Command Palette

Search for a command to run...

Kleines und Effizientes Modell für die Verallgemeinerung der Kantenerkennung

Soria Xavier ; Li Yachuan ; Rouhani Mohammad ; Sappa Angel D.

Zusammenfassung

Die meisten hochwertigen Computer-Vision-Aufgaben basieren auf niedrigstufigen Bildoperationen als ihren anfänglichen Prozessen. Operationen wie Kantenerkennung, Bildverbesserung und Super-Resolution bilden die Grundlage für höhere Bildanalyse. In dieser Arbeit befassen wir uns mit der Kantenerkennung unter Berücksichtigung von drei Hauptzielen: Einfachheit, Effizienz und Generalisierung, da aktuelle state-of-the-art (SOTA) Modelle zur Kantenerkennung in Komplexität gesteigert wurden, um die Genauigkeit zu verbessern. Um dies zu erreichen, stellen wir den Tiny and Efficient Edge Detector (TEED) vor, ein leichtes Faltungsneuronales Netzwerk mit nur 58.000 Parametern, was weniger als 0,2 % der Parameter von SOTA-Modellen entspricht. Das Training auf dem BIPED-Datensatz dauert weniger als 30 Minuten, wobei jede Epoche weniger als 5 Minuten benötigt. Unser vorgeschlagenes Modell ist einfach zu trainieren und konvergiert bereits in den ersten wenigen Epochen schnell, während die vorhergesagten Kantenkarten scharf und von hoher Qualität sind. Darüber hinaus schlagen wir einen neuen Datensatz vor, um die Generalisierungsfähigkeit der Kantenerkennung zu testen. Dieser Datensatz enthält Beispiele aus gängigen Bildern, die in der Kantenerkennung und Bildsegmentierung verwendet werden. Der Quellcode ist unter https://github.com/xavysp/TEED verfügbar.


KI mit KI entwickeln

Von der Idee bis zum Launch – beschleunigen Sie Ihre KI-Entwicklung mit kostenlosem KI-Co-Coding, sofort einsatzbereiter Umgebung und bestem GPU-Preis.

KI-gestütztes kollaboratives Programmieren
Sofort einsatzbereite GPUs
Die besten Preise

HyperAI Newsletters

Abonnieren Sie unsere neuesten Updates
Wir werden die neuesten Updates der Woche in Ihren Posteingang liefern um neun Uhr jeden Montagmorgen
Unterstützt von MailChimp