opendr-eu / opendr

A modular, open and non-proprietary toolkit for core robotic functionalities by harnessing deep learning
Apache License 2.0
611 stars 94 forks source link

Apply cuDNN init fix to all Object Detectors 2D #469

Closed tsampazk closed 9 months ago

tsampazk commented 10 months ago

It seems that the fix applied to yolov5 webcam demo for the cudnn not initialized error (#466), needs to be applied to all object detectors 2D, as i encountered it when running other demos too.

I removed the fix from the webcam demo and added it to the object_detectors_2d module __init__.py file which seems to solve it for all detectors.