如何在 Colab 上使用 OpenCV 和 GPU?

我正在尝试在 Colab 上使用 YOLOv4 进行人体检测。代码包括以下几行:

import cv2

net =cv2.dnn.readNetFromDarknet(yolo_config_path,yolo_weights_path)

if cuda:
    net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
    net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)

但是,如果我直接在 Colab 上运行此单元格,则会遇到如下错误:

Traceback (most recent call last):
  File "run_yolo.py", line 26, in <module>
    net =cv2.dnn.readNetFromDarknet(yolo_config_path,yolo_weights_path)
cv2.error: OpenCV(4.1.2) /io/opencv/modules/dnn/src/darknet/darknet_io.cpp:554: error: (-212:Parsing error) Unsupported activation: mish in function 'ReadDarknetFromCfgStream'

我运行此单元以避免错误:

%cd /content
!git clone https://github.com/opencv/opencv
!git clone https://github.com/opencv/opencv_contrib
!mkdir /content/build
%cd /content/build
!cmake -DOPENCV_EXTRA_MODULES_PATH=/content/opencv_contrib/modules  -DBUILD_SHARED_LIBS=OFF  -DBUILD_TESTS=OFF  -DBUILD_PERF_TESTS=OFF -DBUILD_EXAMPLES=OFF -DWITH_OPENEXR=OFF -DWITH_CUDA=ON -DWITH_CUBLAS=ON -DWITH_CUDNN=ON -DOPENCV_DNN_CUDA=ON /content/opencv
!make -j8 install

当我写下这一行时:

import cv2
cv2.__version__

结果是 4.5.1-dev。

但是当我运行代码时,cv2 是 OpenCV(4.1.2)。我怎么解决这个问题?

stack overflow How to use OpenCV with GPU on Colab?
原文答案

答案:

作者头像

如果要使用Yolov4或其他内容,请使用以下版本的OPENCV:

!pip install opencv-contrib-python==3.4.13.47 --force-reinstall