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

-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,
},
}