Fetch requirements directly from openvino tools (#9154)

* Fetch requirements directly from accuracy_checker

* get requirements from pot & benchmark tool directly
This commit is contained in:
Sergey Lyubimtsev authored and GitHub committed 2021-12-20 14:01:17 +03:00
1 parent f3dd39e3d2
commit ff330ea41e
3 files changed
+12 -32

No files matched your search

+9 -4
View File
@@ -8,14 +8,19 @@ Use this script to create a wheel with OpenVINO™ Python* tools:
$ python setup.py sdist bdist_wheel
"""
import pkg_resources
from setuptools import setup, find_packages
with open('README.md', 'r', encoding='utf-8') as f:
long_description = f.read()
with open('requirements.txt') as f:
required = f.read().splitlines()
required.extend(['openvino'])
with open('requirements.txt') as requirements_txt:
reqs = [
str(requirement)
for requirement
in pkg_resources.parse_requirements(requirements_txt)
]
setup(
name='benchmark_tool',
@@ -37,6 +42,6 @@ setup(
'Operating System :: OS Independent',
],
packages=find_packages(),
install_requires=required,
install_requires=reqs,
python_requires='>=3.6',
)
-28
View File
@@ -1,28 +0,0 @@
defusedxml>=0.7.1
scipy~=1.5.4
jstyleson~=0.0.2
numpy>=1.16.6,<1.20
addict>=2.4.0
pandas~=1.1.5
hyperopt~=0.1.2
networkx~=2.5
tqdm>=4.54.1
texttable~=1.6.3
py-cpuinfo>=7.0.0
PyYAML>=5.4.1
pillow>=8.1.2
scikit-image>=0.17.2
scikit-learn>=0.24.1
yamlloader>=0.5
shapely>=1.7.1
nibabel>=3.2.1
pydicom>=2.1.2
sentencepiece>=0.1.95
tokenizers>=0.10.1
editdistance>=0.5.3
parasail>=1.2.4
fast-ctc-decode>=0.2.5
rawpy>=0.16.0
nltk>=3.5
opencv-python==4.5.*
progress>=1.5
+3
View File
@@ -40,12 +40,14 @@ PKG_INSTALL_CFG = {
'black_list': [],
'prefix': 'benchmark_tool',
'extract_entry_points': True,
'extract_requirements': True,
},
"accuracy_checker": {
'src_dir': OPENVINO_DIR / 'thirdparty' / 'open_model_zoo' / 'tools' / 'accuracy_checker', # noqa:E501
'black_list': ['*tests*'],
'prefix': 'accuracy_checker',
'extract_entry_points': True,
'extract_requirements': True,
},
"omz_tools": {
'src_dir': OPENVINO_DIR / 'thirdparty' / 'open_model_zoo' / 'tools' / 'model_tools', # noqa:E501
@@ -60,6 +62,7 @@ PKG_INSTALL_CFG = {
'black_list': ['*tests*'],
'prefix': 'pot',
'extract_entry_points': True,
'extract_requirements': True,
},
}