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:
1 parent
f3dd39e3d2
commit
ff330ea41e
3 files changed
+12
-32
No files matched your search
@@ -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',
|
||||
)
|
||||
@@ -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
|
||||
@@ -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,
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user