在Jupyter上没有GPU的Darkflow -Notebook-需要简单的代码

# 导入必要的库
import cv2
from darkflow.net.build import TFNet

# 配置Darkflow
options = {
    'model': 'cfg/yolo.cfg',  # 模型配置文件路径
    'load': 'bin/yolov2.weights',  # 模型权重文件路径
    'threshold': 0.5,  # 目标检测的置信度阈值
}

# 创建Darkflow模型
tfnet = TFNet(options)

# 加载图像
img = cv2.imread('image.jpg')

# 进行目标检测
results = tfnet.return_predict(img)

# 在图像上绘制检测结果
for result in results:
    tl = (result['topleft']['x'], result['topleft']['y'])
    br = (result['bottomright']['x'], result['bottomright']['y'])
    label = result['label']
    confidence = result['confidence']
    cv2.rectangle(img, tl, br, (0, 255, 0), 2)
    cv2.putText(img, f"{label}: {confidence:.2f}", tl, cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)

# 显示结果图像
cv2.imshow('Darkflow Result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()