cvlib - cvlib detect_common_objects 不工作

Question

https://github.com/arunponnusamy/cvlib/tree/f46a22f8e13bc4b0267b9057fa88de170d41f96a

我正在检查 cvlib 的示例，并在 spyder Python 3.7 上自行尝试人脸检测工作得很好。性别检测工作得很好，但对象检测不起作用......输出是图像本身。我已经尝试过 BGR2RGB，但没有任何反应。

这是代码

import cvlib as cv
from cvlib.object_detection import draw_bbox 
import cv2
path = r'C:\Users\user\desktop\cod\object_detection_input.jpg'
image = cv2.imread(path)
bbox, label, conf = cv.detect_common_objects(image)
print(bbox, label, conf)
out = draw_bbox(image, bbox, label, conf)
cv2.imshow("object_detection", out)
cv2.waitKey()
cv2.destroyAllWindows()

Answer 1

如果函数返回空列表，可能的原因是训练模型的 yolov3.cfg、yolov3.weights 文件有问题（yolov3.weights 是 242 mg 文件）。

在 Windows 中，这些文件应该在文件夹中：

C:\Users\USER_NAME\.cvlib\object_detection\yolo\yolov3 

文件在功能第一次工作时下载。如果文件损坏，该功能将不起作用但也不会抛出错误！

修复：删除这些文件并再次运行该功能，您将看到两个文件的下载。

Answer 2

我猜是安装库后内核重启的问题。当我重新启动计算机时。有效。