Поворот документів в книжкову орієнтацію з використанням ML

Цей проект розроблений з метою автоматизації процесу повороту документів в книжкову орієнтацію, використовуючи машинне навчання для визначення кута обертання від -30 до 30 градусів. За допомогою цієї ML моделі, ви зможете автоматично вирівнювати текстові документи та зображення для подальшого аналізу або обробки.

Вимоги перед запуском

Перед тим як ви зможете використовувати цей проект, вам потрібно встановити наступне програмне забезпечення:

Python (рекомендована версія 3.10) Встановіть необхідні бібліотеки Python, використовуючи requirements.txt: pip install -r requirements.txt

Необхідно сформувати свої тестові фотографії документів або використати стандартні (з папки test) та підставити їх в папку test та запустити всі клітини з файлу class_creation.ipynb Також необхідно побудувати модель машинного навчання - final_model з файлу model_build.ipynb

Використання

Запустіть програму, яка обробляє документи та визначає їх кут обертання: python main.py --input /шлях/до/папки/з/зображеннями Де /шлях/до/папки/з/зображеннями - це шлях до вашої папки з вхідними документами. Програма автоматично знайде всі зображення форматів .png в папці та її підпапках та обробить їх. Модель автоматично визначить кут обертання та вирівняє документи в книжкову орієнтацію.