AdvancedEAST高效場景文本檢測(附Github地址)

  • 2019 年 12 月 27 日
  • 筆記

AdvancedEAST

AdvancedEAST是一種用於場景影像文本檢測的演算法,主要基於 EAST: An Efficient and Accurate Scene Text Detector,並且還進行了重大改進,使長文本預測更加準確。

(歡迎關注「我愛電腦視覺」公眾號,一個有價值有深度的公眾號~)

優點

  • 基於Keras,易於閱讀和運行
  • 基於EAST,一種先進的文本檢測演算法
  • 易於訓練模型
  • 進行了重大改進,長文本預測更準確。(請參閱下面的「演示結果」部分,並注意激活影像,以黃色網格開頭,以綠色網格結束。)

這份程式碼,本來是作者參加阿里巴巴的文本定位比賽時用到的,後來好心的作者開源了,不僅包括測試,還包括訓練程式碼,請大家記得手動為這位大佬點贊!

在作者的實驗中,AdvancedEast獲得了比East更好的預測準確性,特別是在長文本上。這是因為原始East使用所有像素的預測頂點坐標的加權平均值計算最終頂點坐標,而從四邊形的另一側預測2個頂點太困難了。

network arch

AdvancedEast

EAST

檢測示例:

項目地址:

https://github.com/huoyijie/AdvancedEAST