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

Question

我无法设置和运行一个简单的暗流程序。Infact 甚至无法配置暗流库：

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）上配置暗流并运行上面的代码

Answer 1

通过在 darkflow 目录中的 ipynb 文件中创建文件（从 gi​​thub 下载）并从笔记本执行以下命令来修复：

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