gudbrandtandberg / chessvision Goto Github PK

Hi,

Great job !
Everything works fine for my dataset.
But I am wondering if you have some ideas on how to improve find_quadrangle function.
Sometimes I have really good board mask but this one fails ( no contour found).

Simon

Hello!

• I don't see any pre-trained models available. If they are available for download, how do you download them?
• Note that install.sh is mentioned in the README but does not exist.
  These are separate bugs that need to be fixed. So, for now I'm trying to retrain the model. I started with the square classifier. I'm running

cd ChessVision/ml/training
python train_square_classifier.py

This fails on the quilt package not being found

Traceback (most recent call last):
File "/Users/olivne/oren/ChessVision/ml/training/train_square_classifier.py", line 10, in
from dataset_utils import get_validation_generator, get_training_generator, inverse_labels, get_class_weights, install_data
File "/Users/olivne/oren/ChessVision/ml/training/dataset_utils.py", line 13, in
pieces = quilt.load(PIECES_PATH, hash="d28b23f6aa44126b23150d4108c1af7219f33fbbaa2f61b1c9152a9864f1c8dd")
File "/Users/olivne/miniconda3/envs/chessvision/lib/python3.9/site-packages/quilt/tools/command.py", line 1355, in load
node, pkgroot, info = _load(pkginfo, hash)
File "/Users/olivne/miniconda3/envs/chessvision/lib/python3.9/site-packages/quilt/tools/command.py", line 1346, in _load
raise CommandException("Package {package} not found.".format(package=package))
quilt.tools.command.CommandException: Package gudbrandtandberg/chesspieces not found.

Could you please advise? Thanks.

Nice work for chess vision detection! I want to know can your chessboard detection U-net work well on Youtube videos? I want to detect the chessboard in Youtube video screenshot but most open-sourced chess vision projects fail to do that, because the noise in the image is large (things other than the chessboard) and they cannot identify where the chessboard is.

In addition, I think the current README's instruction is out of date so currently how can I run this project locally? Also the trained model checkpoints seem missing here, are you willing to share them with me? Thank you for your time!

I wanted to try out ChessVision but could not get it running.

python cv_endpoint.py

IOError: Unable to open file (unable to open file: name = '/path/to/ChessVision/weights/best_weights_square.hdf5', errno = 2, error message = 'No such file or directory', flags = 0, o_flags = 0)

I saw you included this in your gitignore:
weights/*

Maybe adding the weights to the source would resolve this error?