daniil-berg/ccaptchas
1
0
Fork 0
generated from daniil-berg/boilerplate-py
Code Issues Pull Requests Projects Releases Wiki Activity

make matplotlib an optional requirement

Browse Source
This commit is contained in:
Daniil Fajnberg 2022-05-16 08:43:40 +02:00
parent 18870a3786
commit a8636822c9
4 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements/common.txt
View File

@ -1,2 +1 @@
matplotlib
numpy numpy

2
requirements/plt.txt Normal file
View File

@ -0,0 +1,2 @@
-r common.txt
matplotlib

5
setup.cfg
View File

@ -1,6 +1,6 @@
[metadata] [metadata]
name = ccaptchas name = ccaptchas
version = 0.1.0 version = 0.1.1
author = Daniil Fajnberg author = Daniil Fajnberg
author_email = mail@daniil.fajnberg.de author_email = mail@daniil.fajnberg.de
description = Character CAPTCHA Solver description = Character CAPTCHA Solver
@ -19,12 +19,13 @@ package_dir =
packages = find: packages = find:
python_requires = >=3.9 python_requires = >=3.9
install_requires = install_requires =
matplotlib
numpy numpy
[options.extras_require] [options.extras_require]
dev = dev =
coverage coverage
plt =
matplotlib
[options.packages.find] [options.packages.find]
where = src where = src

2
src/ccaptchas/infer.py
View File

@ -11,7 +11,6 @@ from keras.api._v2.keras.backend import ctc_decode
from .config import CONFIG from .config import CONFIG
from .preprocess import process_image, decode_label, find_image_files, get_lookup_table from .preprocess import process_image, decode_label, find_image_files, get_lookup_table
from .types import PathT, ImgT, Array from .types import PathT, ImgT, Array
from .visualize import plot_images
def process_predictions(predictions: tf.Tensor) -> tf.Tensor: def process_predictions(predictions: tf.Tensor) -> tf.Tensor:
@ -61,6 +60,7 @@ def load_and_infer(images: Sequence[ImgT], model_dir: PathT, plot_results: bool
model, backward_lookup = load_inference_model(model_dir) model, backward_lookup = load_inference_model(model_dir)
images, labels = predict_and_decode(images, model, backward_lookup) images, labels = predict_and_decode(images, model, backward_lookup)
if plot_results: if plot_results:
from .visualize import plot_images
per_plot = 24 per_plot = 24
for i in range(0, len(images), per_plot): for i in range(0, len(images), per_plot):
plot_images(images[i:(i + per_plot)], labels=labels[i:(i + per_plot)]) plot_images(images[i:(i + per_plot)], labels=labels[i:(i + per_plot)])