/neural-chessboard

♔ An Extremely Efficient Chess-board Detection for Non-trivial Photos ♔

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Computer Vision! Machine learning! A E S T H E T I C!

Neural Chessboard

An Extremely Efficient Chess-board Detection for Non-trivial Photos


We present a set of algorithms that can be used to locate and crop the chess-board/chess-pieces from the picture, including every rectangular grid with any pattern. Our method is non-parametric, and thus does not require the prior knowledge from computer vision and machine learning, which is instead inferred from data. We illustrate the application of our method to a variety of examples, such as chess-board cropping and regular grid-pattern localization. In addition, we present two independent algorithms: PAMG (vertices detector) and FAPL (thermal lines) that can be widely used for other tasks in computer vision.


Files:

$ # basic
$ python3 detect <filename>          # detect and crop chess-board from a photo
$ python3 fen <filename>             # generate FEN from a cropped photo
$ # utils
$ python3 dataset                    # prepare a dataset (for detector)
$ python3 train                      # learn PAMG (neural network)

Dependencies:

TODO:


Raw Process:

BONUS (old gif):