search

Yuliang-Liu / Curve-Text-Detector

This repository provides train＆test code, dataset, det.&rec. annotation, evaluation script, annotation tool, and ranking.
642 stars 156 forks source link

detect rotate rectangles #19

Closed ccnankai closed 6 years ago

ccnankai commented 6 years ago

Hi @Yuliang-Liu I trained simple data. the detection results in the horizontal direction is very good,but in the direction of rotation, few text can be detected . Can this algorithm locate the rotating text (all rectangle,no curve text) 152

0000064r

Yuliang-Liu commented 6 years ago

Yes, it does. I had tried on MSRA-TD500 before, and it went well. Several things may be helpful:

  1. Make sure you have rotating training data.
  2. Using loose NMS threshold (e.g., 0.8) in RPN stage to make sure the horizontal anchors not being prematurely suppressed.
  3. Using PNMS in the second stage, and thus different text lines would not interfere with each other.
  4. (Extended) Make use of pixel-wise prediction (segmentation), e.g., adopting mask branch following Mask RCNN.

Regards, yl

ccnankai commented 6 years ago

Thank you for your advice, thank you.