Skip to content

Pytorch Model to find a skew angle of an image and then de-skew it

Notifications You must be signed in to change notification settings

Mikkicon/ml-deskew

Repository files navigation

train_pool1

TASK - Create a process for turning documents into book orientation

  • Using an ML model that will determine the angle from -30 to 30 degrees.
  • Libraries: pytorch and tf and any supporting libraries.
  • Time constraint: the model should run for <5 seconds, but this is not a critical requirement.
  • Document the process sufficiently so it's and understandable during the review.
  • requirements.txt file, which lists all the libraries used in the process with their versions
  • deadline 11.09.2023

Usage

  1. docker compose up -d
  2. docker exec -it deskew bash
  3. python src/train.py
  4. python src/deskew.py

Check ./output directory for results

About

Pytorch Model to find a skew angle of an image and then de-skew it

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published