Update 13 Dec 2021 : Cuộc thi chính thức đã kết thúc tuy nhiên các bạn vẫn có thể tải dataset và thử nghiệm các giải pháp cho cuộc thi tại https://aihub.vn/competitions/1.
Update 02 Dec 2021 : Team mình (SDSV-AICR) đã đạt top 1 cuộc thi MC-OCR 2021!
Update 23 Nov 2021 : Mình sẽ thuyết trình cho bài báo "MC-OCR Challenge 2021: End-to-end system to extract key information from Vietnamese Receipts" tại hội thảo RIVF2021 vào lúc 17:10 - 17:30 ngày 02/12/2021. Link
Update 23 Apr 2021 : Paper của team mình (SDSV-AICR) "MC-OCR Challenge 2021: End-to-end system to extract key information from Vietnamese Receipts" đã được accept tại hội thảo RIVF2021! Link
Cuộc thi MC_OCR 2021 bao gồm 2 tasks là (1) Đánh giá chất lượng hóa đơn và (2) trích xuất thông tin quan trọng từ ảnh chụp hóa đơn bán lẻ ở Việt Nam. Chi tiết hơn các bạn có thể xem tại trang chủ cuộc thi: https://www.rivf2021-mc-ocr.vietnlp.com/ . Dưới đây là một vài ảnh mẫu
Các đội tham gia chỉ có 1 tháng để làm. Ngoài ra BTC không cho phép gán nhãn bằng tay, không sử dụng dữ liệu ngoài và phải đăng ký pretrained models với tất cả các tasks, có lẽ là để đảm bảo công bằng cho tất cả các đội
Dưới đây là các bước team mình đã thực hiện để giải quyết bài toán này
Environments
- Python 3.6
- Cuda toolkit 10.2
Clone source code
git clone https://github.com/ndcuong91/MC_OCR.git
cd MC_OCR
Cài đặt
pip3 install --upgrade pip
pip3 install -r requirements.txt
pip3 install -e .
cd mc_ocr/text_detector/PaddleOCR
pip3 install -e .
python3 -m pip install paddlepaddle==2.0rc1 -i https://mirror.baidu.com/pypi/simple
cd mc_ocr/text_classifier/vietocr
pip3 install -e .
Trainning: Chuẩn bị dữ liệu training giống như file image_quality_evaluation/train.txt và chạy
cd image_quality_evaluation
python3 train.py --train_mode=1 --path_file=train.txt --img_size=640
Model đạt kết quả cao nhất của team mình đạt được tại epoch 170
Evaluate:
Tạo file image_quality_evaluation/private_test.txt và chạy
cd image_quality_evaluation
python3 train.py --train_mode=0 --path_file=private_test.txt --img_size=640 --path_pretrain=path_to_model --path_save_result=result.txt
Kết qủa cuối cùng sẽ được lưu trong file result.txt
Dươí đây là các bước chuẩn bị dữ liệu và huấn luyện. Đầu tiên, các bạn hãy tải dataset của BTC về và tạo symbolic links
cd mc_ocr/data
ln -s [your_downloaded_train_image_folder] mc_ocr_train
ln -s [your_downloaded_private_test_image_folder] mc_ocr_private_test
Tập dữ liệu training do BTC đưa có có tổng cộng 1155 ảnh. Tuy nhiên team mình thấy nhiều ảnh trong số đó chưa đạt yêu cầu như gắn nhãn sai, nhầm thông tin... nên mình có dùng 1 số rules để lọc chúng đi.
cd mc_ocr/EDA
python3 filter_training_data_by_rules.py
Dữ liệu training sau khi lọc sẽ còn 1090 ảnh.
Tiếp theo bên mình có lọc ra danh sách tên cửa hàng + địa chỉ từ dữ liệu training.
cd mc_ocr/EDA
python3 get_store_dict.py
Kết quả sã là file final_data.json chứa danh sách các cửa hàng (tên và địa chỉ). Danh sách này sẽ được dùng để sửa lỗi cho phần OCR
Hệ thống của team mình có 5 bước chính giống như hình vẽ dưới đây:
Trước hết, các bạn hãy sửa biến sau trong file mc_ocr/config.py
dataset = 'mc_ocr_train_filtered'
Bước này sẽ tìm vị trí của vùng chữ trên ảnh. Team mình sử dụng pre-trained từ PaddleOCR mà không finetune lại gì cả. Các bạn hãy download pre-trained từ link này, giải nén và chỉnh sửa đường dẫn đến pre-trained trong file mc_ocr/config.py
det_model_dir = [your extracted folder]
Bước này sẽ làm nhiệm vụ xoay lại hóa đơn cho thẳng. Bên mình có tạo dữ liệu và training từ đầu cho bước này. Feature extractor được sử dụng là Mobilenetv3.
Đầu tiên các bạn hãy chạy phần text detector
cd mc_ocr/text_detector/PaddleOCR
python3 tools/infer/predict_det.py
Sau đó lọc các ảnh bị ngược hoặc xoay ngang trong tập train (sử dụng confidence của text classify để lọc, với threshold là 0.7)
cd mc_ocr/rotation_corrector
python3 process_mc_ocr_data.py
Tạo synthetic data và augmentation real data từ dữ liệu ở trên:
cd mc_ocr/rotation_corrector
python3 data_process.py
Cuối cùng, sửa line 10, 11 file mc_ocr/rotation_corrector/experiments/mobilenetv3_filtered_public_train.yaml theo đường dẫn từ bước 2. (base_output_dir) để training
cd mc_ocr/rotation_corrector
python3 train_config.py --cfg experiments/mobilenetv3_filtered_public_train.yaml
Pre-trained team mình sử dụng đã để sẵn trong mc_ocr/rotation_corrector/weights
Sau khi xoay lại hóa đơn sẽ vẫn còn nhiều dòng chữ bị nghiêng, bước này sẽ cắt vùng chữ đó và xoay lại cho thẳng để phần OCR được tốt hơn. Team mình chỉ sử dụng xử lý ảnh cơ bản cho phần này
Đây chính là bước OCR đọc chữ từ vùng ảnh đã được xoay từ trên. Team mình sử dụng pre-trained từ open soucre nổi tiếng Vietocr. Rất cảm ơn bạn pbcquoc đã public một model rất tốt ra cộng đồng. Các bạn hãy download pretrained của mô hình vgg19_seq2seq tại đây và chỉnh sửa đường dẫn trong file config.py
cls_model_path = [your downloaded model]
Bước này sử dụng PICK model của tác giả wenwenyu để trích xuất thông tin từ hóa đơn. Team mình có tạo dữ liệu và huấn luyện mô hình từ đầu cho bước này.
Đầu tiên, hãy sửa các biến sau trong file mc_ocr/config.py:
dataset = 'mc_ocr_train_filtered'
det_visualize = False
rot_visualize = False
cls_visualize = False
Sau đó, chạy lần lượt các bước text detector, rotation corrector và text classifier
cd mc_ocr/text_detector/PaddleOCR
python3 tools/infer/predict_det.py
cd mc_ocr/rotation_corrector
python3 inference.py
cd mc_ocr/text_classifier
python3 pred_ocr.py
Dùng model thu được ở bước 2. rotation_corrector để chỉnh sửa file csv:
cd mc_ocr/rotation_corrector
python3 rotate_csv.py
Sau đó chuẩn bị dữ liệu training
cd mc_ocr/key_info_extraction
python3 create_train_data.py
Bước trên sẽ tạo ra 2 file train.csv và val.csv trong thư mục ... Tiếp theo các bạn hãy chỉnh sửa đường dẫn đến 2 file trên trong file key_info_extraction/PICK/config.json (line 59 và 74)
Cuối cùng sẽ là bước training
cd mc_ocr/key_info_extraction/PICK
bash run.sh
Kết quả của bước training sẽ là file model_best.pth nằm trong thư mục key_info_extraction/PICK/saved/models/PICK_Default/test... Các bạn hãy chỉnh sửa đường dẫn đến file đó trong config.py
kie_model = [your trained model]
model mà team mình train được có thể download tại đây
Chuyển dataset sang private test trong file mc_ocr/config.py
dataset = 'mc_ocr_private_test'
Sau đó chạy lần lượt các bước sau
text detector
cd mc_ocr/text_detector/PaddleOCR
python3 tools/infer/predict_det.py
rotation corrector
cd mc_ocr/rotation_corrector
python3 inference.py
text classifier
cd mc_ocr/text_classifier
python3 pred_ocr.py
key information extraction
cd mc_ocr/key_info_extraction/PICK
python3 test.py
submission
cd mc_ocr/submit
python3 submit.py
Kết quả cuối cùng là file submit/mc_ocr_private_test/result.csv.
File submit cuối cùng của team mình trong cuộc thi là submit/mc_ocr_privater_test/results_2301_3_merged.zip
Dưới đây là 1 vài kết quả sau khi visualize:
| Task | Score | Top | Speed |
|---|---|---|---|
| 1. Quality evaluation | RMSE: 0.12 | 3 | 2ms (GPU) |
| 2. Key information extraction | CER: 0.23 | 2 | 1.5s |
GPU team mình sử dụng là Nvidia RTX 2080 Ti
- PaddleOCR: https://github.com/PaddlePaddle/PaddleOCR
- PICK-Pytorch: https://github.com/wenwenyu/PICK-pytorch
- Vietocr: https://github.com/pbcquoc/vietocr
Trong đó team mình sử dụng pre-trained từ PaddleOCR và Vietocr
- Dữ liệu từ BTC khá lớn (tổng cộng 2k ảnh, với 1155 ảnh đã được gán nhãn). Tuy nhiên, khá nhiều hóa đơn là từ MINIMART ANAN và VinCommerce nên sự đồng đều và đa dạng là không cao.
- Dữ liệu đã gán nhãn có tỉ lệ gán nhãn lỗi khoảng 5-10%. Tỉ lệ này là chấp nhận được với một bộ dataset thực tế.
- Yêu cầu không sử dụng dữ liệu ngoài với không gán nhãn thủ công khiến cho kết quả đạt được không quá tốt.
- cuongnd (nguyenduycuong2004@gmail.com)
- anhnt
- chungnx