Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion tools/accuracy_checker/requirements-core.in
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
# core components
defusedxml>=0.7.1
numpy>=1.16.6,<2.1.0
numpy>=1.16.6,<2.1.0;python_version<"3.12"
numpy>=1.26.0,<2.3.0;python_version>="3.12"
openvino-telemetry>=2023.2.1
PyYAML>=5.4.1
pillow>=8.1.2
packaging
14 changes: 10 additions & 4 deletions tools/accuracy_checker/requirements-extra.in
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,12 @@
tqdm>=4.54.1

# image reading and preprocessing
opencv-python==4.5.4.60;python_version<"3.12"
opencv-python==4.10.0.84;python_version>="3.12"
# nncf's upperbound: https://github.com/openvinotoolkit/nncf/blob/1de67217adf60a29afa310edf4a5f4d36314446a/setup.py#L104
networkx<3.6
scikit-image>=0.19.2
scikit-image==0.18.3;python_version<"3.12"
scikit-image==0.24.0;python_version>="3.12"

imagecodecs~=2022.2.22;python_version<="3.10"
imagecodecs~=2024.6.1;python_version>="3.11"
Expand Down Expand Up @@ -38,7 +41,6 @@ nltk>=3.5
# DNA sequence matching
parasail>=1.2.4;platform_system!="Windows"
parasail~=1.2.4;platform_system=="Windows"
fast-ctc-decode>=0.2.5;python_version<"3.11"

# raw image formats processing
rawpy>=0.17.1
Expand All @@ -47,7 +49,11 @@ rawpy>=0.17.1
lmdb>=1.2.1

# pandas datasets support
pandas>=1.1.5,<2.1
pandas>=1.1.5,<2.3

# word-based representations of numbers
inflect>=7.4.0
inflect>=7.3.0

# visualization
matplotlib==3.8.4;python_version<"3.12"
matplotlib==3.9.3;python_version>="3.12"
13 changes: 9 additions & 4 deletions tools/accuracy_checker/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import warnings
import platform
import subprocess # nosec B404 # disable import-subprocess check
from distutils.version import LooseVersion
from packaging.version import Version
from pathlib import Path
from setuptools import find_packages, setup # pylint:disable=W9902
from setuptools.command.test import test as test_command # pylint:disable=W9902
Expand Down Expand Up @@ -55,7 +55,7 @@ def read(*path):
def check_and_update_numpy(min_acceptable='1.15'):
try:
import numpy as np # pylint:disable=C0415
update_required = LooseVersion(np.__version__) < LooseVersion(min_acceptable)
update_required = Version(np.__version__) < Version(min_acceptable)
except ImportError:
update_required = True
if update_required:
Expand Down Expand Up @@ -106,6 +106,12 @@ def prepare_requirements():
check_and_update_numpy()



_extras = _extras + ['pycocotools>=2.0.2', 'torch>=0.4.0', 'torchvision>=0.2.1', 'lpips', 'soundfile', "torchmetrics", "diffusers"]
if sys.version_info.major == 3 and sys.version_info.minor < 12:
_extras.append('crf_beam;platform_system=="Linux"')
_extras.append('kenlm @ git+https://github.com/kpu/kenlm.git@4cb443e60b7bf2c0ddf3c745378f76cb59e254e5#egg=kenlm')

setup(
name="accuracy_checker",
description="Deep Learning Accuracy validation framework",
Expand All @@ -121,6 +127,5 @@ def prepare_requirements():
install_requires=_requirements,
tests_require=[read("requirements-test.in")],
cmdclass={'test': PyTest, 'install_core': CoreInstall},
extras_require={'extra': _extras + ['pycocotools>=2.0.2', 'crf_beam;platform_system=="Linux"', 'torch>=0.4.0', 'torchvision>=0.2.1', 'lpips', 'soundfile', "torchmetrics", "diffusers",
'kenlm @ git+https://github.com/kpu/kenlm.git@4cb443e60b7bf2c0ddf3c745378f76cb59e254e5#egg=kenlm']}
extras_require={'extra': _extras}
)
Loading