Detecção de faces com CNN e Dlib

[Julio Francisco de Lima]

Bom dia.

Estou recebendo este erro na parte de detecção de faces com CMM e Dlib.

RuntimeError Traceback (most recent call last)

<ipython-input-60-df4c6f21ccc5> in <module>()
—-> 1 deteccoes = detector_face_cnn(imagem, 1)
2 for face in deteccoes:
3 l, t, r, b, c = face.rect.left(), face.rect.top(), face.rect.right(), face.rect.bottom(), face.confidence
4 print(c)

RuntimeError: Error while calling cudnnConvolutionBiasActivationForward( context(), &alpha1, descriptor(data), data.device(), (const cudnnFilterDescriptor_t)filter_handle, filters.device(), (const cudnnConvolutionDescriptor_t)conv_handle, (cudnnConvolutionFwdAlgo_t)forward_algo, forward_workspace, forward_workspace_size_in_bytes, &alpha2, out_desc, out, descriptor(biases), biases.device(), identity_activation_descriptor(), out_desc, out) in file /tmp/pip-install-uwm3abyj/dlib_961ff087af364da4874d302ad406612d/dlib/cuda/cudnn_dlibapi.cpp:1237. code: 9, reason: CUDNN_STATUS_NOT_SUPPORTED

 

[Gabriel Alves]

Olá Julio!

Esse problema pode ocorrer dependendo de qual GPU foi atribuída em sua sessão do Colab. Na verdade, foi devido a uma mudança muito recente que esse erro começou a acontecer, mais especificamente após uma atualização do Dlib, que gerou um conflito com a GPU utilizada pelo Colab. Portanto, a solução é utilizar uma versão anterior àquela em que ocorre o problema. Escolhi a 19.22.1, que é uma versão recente também (e antes daquela que começou a dar esse problema) então se você usar essa versão o erro não ocorrerá.

Então, para resolver esse erro você precisa executar antes de import dlib a linha abaixo:

!pip install dlib==19.22.1
em seguida, execute import dlib. Se você já importou o dlib nesta sessão, basta reiniciar o ambiente de execução (selecione Runtime > Restart runtime).

A melhor solução acredito ser essa, até o problema ser resolvido em alguma versão futura.

Qualquer coisa, veja aqui o Colab da aula atualizado.

[Autor]

Posts