huyhoang17 / kuzushiji_recognition

[Late Submission] Solution for Kuzushiji recognition (Kaggle competition)

Home Page:https://viblo.asia/p/V3m5WPngKO7

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Kuzushiji Recognition

[Late Submission] Solution for kuzushiji recognition (kaggle competition)

Segmentation model

  • Unet with custom resnet-based backbone
Evaluate on detection model

Classification model

  • Baseline model for kuzushiji character recognition

  • Number of classes: 3422

Command

  • Clone repository
git clone https://github.com/huyhoang17/kuzushiji_recognition
cd kuzushiji_recognition
  • Install some prerequisite libs
pip install -r requirements.txt
  • [Optional] Install Git LFS and pull model files, follow by this tutorial

  • Open kuzu_tfserving.config on editor, change base_path of 2 models to absolute path to each sub-folder

# change this line
base_path: '/home/phan.huy.hoang/workspace/projects/kaggle_kuzushiji/model_server/kuzu_segment'

# to
base_path: '/absolute-path-to-root-folder/model_server/kuzu_segment'
  • Install tensorflow_model_server from this link

  • Run tensorflow model server

tensorflow_model_server --port=8500 --rest_api_port=8501 --model_config_file=/absolute-path-to-kuzu-tfserving.config
  • Test detection & recognition model
python3 src/grpc_infer.py
  • Check result image in assets folder

Result

Result 2

  • I tested on Chữ_Nôm and used pre-trained detection model from kuzushiji dataset

Đại Việt sử ký toàn thư

Chùa Thắng Nghiêm

Lục Vân Tiên

TODO

  • Add pytorch code

  • Update docker / docker-compose

Other issue

Contact

  • If you find this repo useful, please star the project to let people know that it's reliable ⭐⭐⭐ Thank you!

  • For more information, please contact me at email address: hoangphan0710@gmail.com

About

[Late Submission] Solution for Kuzushiji recognition (Kaggle competition)

https://viblo.asia/p/V3m5WPngKO7

License:MIT License


Languages

Language:Python 100.0%