noword / image2sgf

recongnize go game status from a image, output to a sgf file

Dependencies

python >= 3.6

PyTorch

torchvision

cv2

numpy

matplotlib

pillow

sgfmill

pyautogui

for GUI wxPython

Usage:

Download and extract pth files from https://github.com/noword/image2sgf/releases

usage: img2sgf.py [-h] [--sgf_name SGF_NAME] [--capture] [--save_images] [image_name]

positional arguments:
  image_name           input image file name

optional arguments:
  -h, --help           show this help message and exit
  --sgf_name SGF_NAME  output sgf file name
  --capture            capture the screenshot
  --save_images        save grid images

Examples:

img2sgf.py examples\goban.png

GUI

img2sgf_gui.pyw

image2sgf's Issues

发现一个很有趣的问题

把一个棋谱照片quality-100.jpg，通过不同的压缩参数生成quality-30.jpg, quality-50.jpg, quality-60.jpg, quality-70.jpg, quality-80.jpg, quality-90.jpg。发现quality 70以上的可以正常转成相应的sgf，60及以下的都识别有错误的棋子，请noword研究一下如何在底quality下的jpg也可以被正确识别。

下载

非常棒，我在3090ti卡上测试成功了

转换一个图片大概需要8秒时间。不知道怎么能够加快速度？

棋盘右下角点识别错误，造成棋子识别错误

原图

How can i get the 'board_mobile.pth' model

Amazing job! 'board.pth' is very accurate, but a bit large and slow. Can you share your 'board_mobile.pth'?

Fails on image found in the wild

Using the image from a reddit post fails on step one and never completes step 2:

(status bar should be showing "step 2: perspective correct the board, then classify stones")

I cropped the image:

And tried again:

Apart from a few mistakes (that still leave a valid board), it appears to complete step 2, but then gives the error "Recognition failed" and the status bar reads "Error: Can't generate sgf."

error: board.pth is missing

Hi noword,

I get your stone.pth model. However, the board.pth is missing.
Can you let me know where to find the board.pth?

Bests,

Why this style gameboard cannot recgonize

download image here

one stone error

download image here

Missing one white stone:

Can the accuracy be improved by adjusting parameters?

A blank point was misidentified as a white stone.

can't find the imgs.py under /gui directory

Hi author of image2sgf,

This is very useful project. However, I'm looking for help to solve an issue when runing img2sgf.py.

from gui.main import *
import imgs
ModuleNotFoundError: No module named 'imgs'

I'm thinking there might be a imgs.py missing. Or do you have any idea why this is happening. I'm a newbie to python and would love help!

Bests,
Wendy