ipazc / mtcnn

MTCNN face detection implementation for TensorFlow, as a PIP package.
MIT License
2.19k stars 526 forks source link

Error with tf-nightly-gpu 2.5.0.dev20201207 #104

Open fspider opened 3 years ago

fspider commented 3 years ago

I am using this GPU - RTX3090. So installed Cuda 11.1, Cudnn 8.05, tf-nightly-gpu 2.5.0.dev20201207. While using MTCNN, getting this error.

  File "faceDetector.py", line 11, in __init__
    self.detector = MTCNN()
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/mtcnn/mtcnn.py", line 88, in __init__
    self._pnet, self._rnet, self._onet = NetworkFactory().build_P_R_O_nets_from_file(weights_file)
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/mtcnn/network/factory.py", line 123, in build_P_R_O_nets_from_file
    p_net = self.build_pnet()
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/mtcnn/network/factory.py", line 40, in build_pnet
    p_layer = Conv2D(10, kernel_size=(3, 3), strides=(1, 1), padding="valid")(p_inp)
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 75, in symbolic_fn_wrapper
    return func(*args, **kwargs)
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/keras/engine/base_layer.py", line 446, in __call__
    self.assert_input_compatibility(inputs)
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/keras/engine/base_layer.py", line 310, in assert_input_compatibility
    K.is_keras_tensor(x)
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 695, in is_keras_tensor
    if not is_tensor(x):
  File "/home/troica/anaconda3/envs/test_36/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 703, in is_tensor
    return isinstance(x, tf_ops._TensorLike) or tf_ops.is_dense_tensor_like(x)
AttributeError: module 'tensorflow.python.framework.ops' has no attribute '_TensorLike'

Screenshot from 2020-12-10 01-32-52