nsu-ai-team / noise_supression Goto Github PK

Line 58. Change "sckit-learn" to "scikit-learn"

Hi folks,

thanks for this contribution. I'm trying to install it in my computer, but I got the following error when trying to load the pretrained model:
Traceback (most recent call last): File "enhance_audio.py", line 104, in <module> ae = AudioEnhancer() File "enhance_audio.py", line 15, in __init__ self.model = keras.models.load_model(os.path.join(os.path.dirname(__file__), 'model/md_10000iters.h5')) File "/media/diego/disco/software/anaconda3/lib/python3.6/site-packages/keras/models.py", line 233, in load_model model = model_from_config(model_config, custom_objects=custom_objects) File "/media/diego/disco/software/anaconda3/lib/python3.6/site-packages/keras/models.py", line 307, in model_from_config return layer_module.deserialize(config, custom_objects=custom_objects) File "/media/diego/disco/software/anaconda3/lib/python3.6/site-packages/keras/layers/__init__.py", line 54, in deserialize printable_module_name='layer') File "/media/diego/disco/software/anaconda3/lib/python3.6/site-packages/keras/utils/generic_utils.py", line 139, in deserialize_keras_object list(custom_objects.items()))) File "/media/diego/disco/software/anaconda3/lib/python3.6/site-packages/keras/models.py", line 1204, in from_config if 'class_name' not in config[0] or config[0]['class_name'] == 'Merge': KeyError: 0

Could you provide a pretrained model which is compatible with this Keras version or at least could you specify the expected Keras version in the setup file?

Steps:

1. apt install git-lfs
2. git clone https://[email protected]/nsu-ai-team/noise_supression.git
3. python setup.py install

from noise_supression.enhance_audio import AudioEnhancer
ae = AudioEnhancer()

AttributeError: module 'noise_supression' has no attribute 'enhance_audio'

(noise_env) raakhal@raakhal-HP-Pavilion-Laptop-15-cc1xx:~/artivatic/noise_supression/noise_supression$ python enhance_audio.py MER_rohan.wav MER_rohan1.wav
Using TensorFlow backend.
WARNING: Logging before flag parsing goes to stderr.
W0612 14:19:41.057055 140574799079168 deprecation.py:323] From /home/raakhal/artivatic/environments/noise_env/lib/python3.6/site-packages/tensorflow-2.0.0b0-py3.6-linux-x86_64.egg/tensorflow/python/compat/v2_compat.py:65: disable_resource_variables (from tensorflow.python.ops.variable_scope) is deprecated and will be removed in a future version.

if isinstance(identifier, tf.Optimizer):

AttributeError: module 'tensorflow' has no attribute 'Optimizer'
(noise_env) raakhal@raakhal-HP-Pavilion-Laptop-15-cc1xx:~/artivatic/noise_supression/noise_supression$ B