es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: ONNX

No se puede crear una sesión de inferencia en onnxruntime.

No puedo crear una instancia de InferenceSession utilizando onnxruntime. Mi plataforma es Mac OS (Big Sur). El código ni siquiera arroja ninguna excepción. El proceso simplemente se cierra. El mismo código funciona perfectamente en Windows. Aquí están las capturas de pantalla: La captura de pantalla que se muestra a continuación . . . Read more

Error de YOLOv5 OpenCV: (-215:Error de aserción) mientras se utiliza .onnx

Acabo de crear un modelo yolov5 y lo exporté en formato onnx para que sea utilizable con opencv, pero sigo obteniendo el siguiente error: [ERROR:0] global D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp (3554) cv::dnn::dnn4v20211004::Net::Impl::getLayerShapesRecursively OPENCV/DNN: [Reshape]:(466): getMemoryShapes() lanza una excepción. entradas=1 salidas=1/1 blobs=0 [ERROR:0] global D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp (3557) cv::dnn::dnn4v20211004::Net::Impl::getLayerShapesRecursively entrada[0] = [ 1 24 52 52 . . . Read more

onnxruntime no está utilizando CUDA.

Entorno: – CentOS 7 – python 3.9.5 – CUDA: 11.4 – cudnn: 8.2.4 – onnxruntime-gpu: 1.9.0 – controlador nvidia: 470.82.01 – 1 gpu tesla v100 Mientras que onnxruntime parece reconocer la gpu, cuando se crea InferenceSession, parece que ya no reconoce la gpu. El siguiente código muestra este síntoma. import . . . Read more