Skip to content

SeeBook2022/SeeBook-OCR

Repository files navigation

📕 SeeBook-OCR

Object detectionOCR을 활용한 여러 권의 책 정보 검출 및 검색 서비스입니다.

  • SeSAC x Modulabs AIFFEL hackathon3에서 진행한 프로젝트 입니다.
  • 진행 기간: 2022.01.24 ~2022.03.11

seebookocr





✔ Development Environment

  • Google Colab Pro (GPUs: K80, P100, T4)
  • GitHub





✔ Tech

  • YOLOv5
  • Google Cloud Vision API
  • Streamlit
  • Python3
  • Pytorch
  • OpenCV2
  • Pandas
  • Wandb
  • Pillow(PIL)
  • Numpy





✔ Pipeline

Two - Stage Detection 방식을 통한 책 제목 검출 (YOLOv5를 stage1, stage2에서 모두 사용)

pl



pl2





✔ Evaluation

stage1, stage2, stage3 정량 평가 진행 [Link]









✔ Contributors


박진우


장유진


김성연


나융