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

Question

我无法设置和运行一个简单的暗流程序。事实上，我甚至不能配置darkflow库：

from darkflow.net.build import TFNet
==> ModuleNotFoundError: No module named 'darkflow'

我的目标是运行以下程序：

from darkflow.net.build import TFNet
import cv2

options = {"model": "cfg/yolo.cfg", "load": "bin/yolo.weights", "threshold": 0.1}

tfnet = TFNet(options)

imgcv = cv2.imread("./test/dog.jpg")
result = tfnet.return_predict(imgcv)
print(result

请建议步骤，以便我可以在Jupyter Notebook (没有GPU)上配置darkflow并运行上面的代码

Answer 1

已通过在darkflow目录(从github下载)的ipynb文件中创建文件并从笔记本中执行以下命令修复：

!python3 setup.py build_ext --inplace
!pip install -e .
!pip install .