Skip to main content

Pokemon TCG Card Recognizer

Project description

Pokemon Card Recognizer

Recognize a Pokémon Card in an image or video.

from pokemon_card_recognizer.api.card_recognizer import CardRecognizer, OperatingMode

# init and set output paths
recognizer = CardRecognizer(
    mode=OperatingMode.PULLS_VIDEO
)
recognizer.set_summary_file(summary_file="summary.txt")
recognizer.set_output_path(output_path="out_figs")

# run recognizer on video and visualize results
pulls = recognizer.exec("/path/to/video")
recognizer.vis()

Example analysis of a booster pack opening video:

Benchmarks: https://docs.google.com/presentation/d/1Q6PzJpqyyLFvtkLGoCFeXaHYOauOzeF_umx8V3w4fV8/edit?usp=sharing

Installation:

make conda_dev
conda activate card_rec_env
pip install pokemon-card-recognizer

Note that the CardRecognizer works MUCH (~5x-10x) faster on NVIDIA GPU, so it is highly recommended that you have CUDA. If CUDA is available, the CardRecognizer will automatically use it. Otherwise, the CardRecognizer will default to CPU which is substantially slower and not recommended for batch processing.

If processing video files, you may also need to download and install FFMPEG (https://ffmpeg.org/) which is used to uncompress videos into image frames.

On linux:

sudo apt update
sudo apt install ffmpeg

Optional: The default OCR backend used by CardRecognizer is easy_ocr which is installed via the conda, but if you choose to use PyTesseract instead, you will need to install it:

sudo apt install tesseract-ocr
sudo apt install libtesseract-dev  

Example Usage to Recognize a Card in a Single Image:

from pokemon_card_recognizer.api.card_recognizer import CardRecognizer, OperatingMode
recognizer = CardRecognizer(
    mode=OperatingMode.SINGLE_IMAGE,
    set_name="master"
)
pred_result = recognizer.exec("/path/to/image")
detected_card = recognizer.classifier.reference.lookup_card_prediction(
    card_prediction=pred_result[0]
)
print(detected_card.set)
print(detected_card.name)
print(detected_card.number)

Useful Operating Modes for CardRecognizer:

# process a single image
OperatingMode.SINGLE_IMAGE

# process a directory of images
OperatingMode.IMAGE_DIR

# process a video file
OperatingMode.VIDEO

# process a video file where cards are being shown ("pulled") sequentially 
(no assumption on # of cards shown in the video)
OperatingMode.PULLS_VIDEO

# process a video file where cards are being shown sequentially, 
# coming from a booster pack (assumes 11 cards are being shown in the video and 
# finds the best statistical estimation of at most 11 most likely shown cards).
OperatingMode.BOOSTER_PULLS_VIDEO

Card Reference Sets

Note that you can change "set_name" in the CardRecognizer constructor to whatever specific set reference (e.g. "base1", "jungle", etc) you want. For list of supported reference sets, view

from pokemon_card_recognizer.reference.core.build import ReferenceBuild
print(ReferenceBuild.supported_card_sets())

Building Card Reference:

The pypi package and GitHub source for pokemon-card-recognizer comes bundled with pre-rebuilt references for all major Pokémon card sets. It is recommended to use the pre-built references. If however, you want to rebuild the reference for some reason, you can do:

python pokemon_card_recognizer/reference/core/build.py [PTCGSDK_API_KEY]

where PTCGSDK_API_KEY is your PTCGSDK API key. You can get one here: https://pokemontcg.io/.

Note that building and evaluating the reference can take an hour or more, depending on your system configuration. It is recommended to use the prebuilt card references that come pre-bundled.

*This API is unofficial, open-source, in development, and not affiliated with the Pokémon Company.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pokemon_card_recognizer-0.0.1.3.8.7.tar.gz (20.5 MB view details)

Uploaded Source

Built Distribution

File details

Details for the file pokemon_card_recognizer-0.0.1.3.8.7.tar.gz.

File metadata

File hashes

Hashes for pokemon_card_recognizer-0.0.1.3.8.7.tar.gz
Algorithm Hash digest
SHA256 eaa00ed44b9ab75bca3eef49046462a6769ea680b425818fe60110cdb2669be7
MD5 07fceeb4037d737f5c822ab76b268822
BLAKE2b-256 afa02e2f033306802cb1a2e107f734855d93e2f3afbb274be9f67d4790bc9b35

See more details on using hashes here.

File details

Details for the file pokemon_card_recognizer-0.0.1.3.8.7-py3-none-any.whl.

File metadata

File hashes

Hashes for pokemon_card_recognizer-0.0.1.3.8.7-py3-none-any.whl
Algorithm Hash digest
SHA256 d2add3af4818bc0c37c44ab7c7a9dc3c67c8f58a4b0fc331045ee28f9f8f4bcf
MD5 35eec0b8ca99e8fa27d49be166fa8d29
BLAKE2b-256 375b3f48efb7d04514059e1e802097803666343ea78bc117f04e2817dab2accc

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page