diff --git a/tools/pot/openvino/tools/pot/algorithms/quantization/optimization/algorithm.py b/tools/pot/openvino/tools/pot/algorithms/quantization/optimization/algorithm.py index 2ceed76b035..3f9fb1ef808 100644 --- a/tools/pot/openvino/tools/pot/algorithms/quantization/optimization/algorithm.py +++ b/tools/pot/openvino/tools/pot/algorithms/quantization/optimization/algorithm.py @@ -9,7 +9,6 @@ from functools import partial from pathlib import Path import numpy as np -import pandas as pd import scipy.optimize from ..qnoise_estimator.algorithm import QuantNoiseEstimator @@ -166,7 +165,7 @@ class OptimizationAlgorithm(Algorithm): Path('/'.join(self._result_filename.split('/')[:-1])).mkdir( parents=True, exist_ok=True ) - pd.DataFrame(self._results).to_csv(self._result_filename) + np.savetxt(self._result_filename, self._results, delimiter=",", fmt='%s') if self._dump_model_prefix: dump_path = self._dump_model_prefix + '{:05}'.format( self._optimization_iter diff --git a/tools/pot/openvino/tools/pot/algorithms/quantization/qnoise_estimator/algorithm.py b/tools/pot/openvino/tools/pot/algorithms/quantization/qnoise_estimator/algorithm.py index 2e0b52d0c73..1392844c233 100644 --- a/tools/pot/openvino/tools/pot/algorithms/quantization/qnoise_estimator/algorithm.py +++ b/tools/pot/openvino/tools/pot/algorithms/quantization/qnoise_estimator/algorithm.py @@ -4,7 +4,6 @@ from copy import deepcopy import numpy as np -import pandas as pd from openvino.tools.mo.graph.graph import rename_node from .utils import get_composite_model @@ -98,7 +97,7 @@ class QuantNoiseEstimator(Algorithm): 'layer_name': list(stat_calculation_layers.values()), } if 'results_dump_filename' in self._config: - pd.DataFrame(noise_data).to_csv(self._config['results_dump_filename']) + np.savetxt(self._config['results_dump_filename'], noise_data, delimiter=",", fmt='%s') return noise_data def layerwise_fq_noise(self, model): @@ -175,7 +174,7 @@ class QuantNoiseEstimator(Algorithm): noise_data = {'noise_metric': qnoise_values, 'layer_name': node_names} if 'results_dump_filename' in self._config: - pd.DataFrame(noise_data).to_csv(self._config['results_dump_filename']) + np.savetxt(self._config['results_dump_filename'], noise_data, delimiter=",", fmt='%s') return noise_data def get_nonquantized_model(self, model): diff --git a/tools/pot/setup.py b/tools/pot/setup.py index bbee577ec06..ea91d3901b6 100644 --- a/tools/pot/setup.py +++ b/tools/pot/setup.py @@ -65,15 +65,14 @@ if '--install-dev-extras' in sys.argv: sys.argv.remove('--install-dev-extras') INSTALL_REQUIRES = [ + "numpy>=1.16.6", "scipy~=1.7; python_version == '3.7'", "scipy>=1.8; python_version >= '3.8'", "jstyleson>=0.0.2", - "numpy>=1.16.6", "addict>=2.4.0", "networkx<=2.8.8", "tqdm>=4.54.1", "texttable>=1.6.3", - "pandas~=1.3.5", "openvino-telemetry>=2022.1.0" ]