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

时间:2019-04-23 10:11:54

标签: jupyter-notebook gpu darkflow

我无法设置和运行简单的Darkflow程序。实际上甚至无法配置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并运行上面的代码

1 个答案:

答案 0 :(得分:0)

通过在darkflow目录(从github下载)的ipynb文件中创建文件并从笔记本执行以下操作来解决:

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