OCR-yolov5-SwinIR-STARNet (Colab)

OCR(Korean) 한국어 메뉴판 OCR

• 기존 프로젝트 이후 개선 방향

  • 메뉴판 디텍션 이후에 다시 다른 한글 디텍션 (2stage)
    • 느린 속도 Detection 속도로 인해 가벼운 모델 사용 필요
  • 세로로 되어있는 한글 Detection
  • 세로로 되어있는 한글 Recognition
  • 기울어진 메뉴판 인식
  • 성능 향상

Run demo

• Requirement

!pip install timm

• Run demo.py

%cd /content/Korean-OCR-YOLOv5-SwinIR-STARNet
!CUDA_VISIBLE_DEVICES=0 python3 demo.py --image_folder /content/OCR-yolov5-SwinIR-STARNet/img_demo
가중치 수정중

• Result : The result will be saved in './results'

Ex

• 간판 (weight 루트값 수정 필요)

• 메뉴판 Detection

• 메뉴판 end-to-end

Reference

• https://github.com/jingyunliang/swinir
• https://github.com/clovaai/deep-text-recognition-benchmark
• https://github.com/aqntks/Easy-Yolo-OCR
• https://github.com/ultralytics/yolov5