Skip to content
This repository has been archived by the owner on Jan 8, 2024. It is now read-only.

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

License

Notifications You must be signed in to change notification settings

huyhoang17/kuzushiji_recognition

Repository files navigation

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