search

huichen5796 / 2022-studienarbeit-hui-chen

a tool for detecting tables in image and analysing complex header
3 stars 0 forks source link

TableNet #52

Closed huichen5796 closed 2 years ago

huichen5796 commented 2 years ago

image

Quelle des Models: https://github.com/asagar60/TableNet-pytorch

Die Leistung dieses Modells ist bereits sehr gut, ist verfügbar

huichen5796 commented 2 years ago

Ich habe versucht, dieses Modell zu reproduzieren, aber die Ergebnisse sind sehr schlecht, das Problem kann in der Datenverarbeitung und Parameterinitialisierung liegen。

Es dauert eine Minute, um ein Bild auf der CPU zu verarbeiten, und nur 1 Sekunde auf der GPU, also muss ich eine Schnittstelle schreiben, um die GPU von Colab aufzurufen

huichen5796 commented 2 years ago

image

Reproduktion von mir

huichen5796 commented 2 years ago

image Das reproduzierte Modell von mir gibt eine Vorhersage für ein anderes Bild aus, was ist los?

huichen5796 commented 2 years ago

Vielleicht liegt ein Fehler in der Korrespondenz zwischen image und mask vor. Ändert wird die Maskenlesemethode und gelesen wird nun die entsprechende Maske gemäß dem gelesenen Bildpfad vorher: image nun: image

huichen5796 commented 2 years ago

image

jetzt nach nur 10 epochs ist die Prognose schon gut!