Darkflow找不到cfg和weights fi

2024-10-02 18:19:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试在windows机器上设置YOLO在令人兴奋的树冠环境中。由于tensorflow和darkflow似乎在PackageManager上不可用,所以我通过pip安装了它们

pip install tensorflow

在从github下载darkflow目录并从canopy命令提示符导航到它之后

^{pr2}$

然后呢

python setup.py build_ext --inplace

为了建立赛顿实用程序。现在,命令行演示似乎可以正常工作

python flow --model cfg/yolo.cfg --load bin/yolo.weights --demo videofile.mp4 --saveVideo 

但是当我尝试在我的树冠编辑器中运行以下命令时:

from darkflow.net.build import TFNet
option = {'model': 'cfg/yolo.cfg', 'load': 'bin/yolo.weights', 'threshold': 0.3}
tfnet = TFNet(option)

它给了我一个错误:

AssertionError                            Traceback (most recent call last)
c:\users\umair\appdata\local\temp\tmp9n4e92.py in <module>()
      4 option = {'model': 'cfg/yolo.cfg', 'load': 'bin/yolo.weights', 'threshold': 0.3}
      5 
----> 6 tfnet = TFNet(option)

C:\Users\umair\AppData\Local\Enthought\Canopy\edm\envs\User\lib\site-packages\darkflow\net\build.py in __init__(self, FLAGS, darknet)
     56 
     57                 if darknet is None:
---> 58                         darknet = Darknet(FLAGS)
     59                         self.ntrain = len(darknet.layers)
     60 
C:\Users\umair\AppData\Local\Enthought\Canopy\edm\envs\User\lib\site-packages\darkflow\dark\darknet.py in __init__(self, FLAGS)
     11 
     12     def __init__(self, FLAGS):
---> 13         self.get_weight_src(FLAGS)
     14         self.modify = False
     15 
C:\Users\umair\AppData\Local\Enthought\Canopy\edm\envs\User\lib\site-packages\darkflow\dark\darknet.py in get_weight_src(self, FLAGS)
     45         else:
     46             assert os.path.isfile(FLAGS.load), \
---> 47             '{} not found'.format(FLAGS.load)
     48             self.src_bin = FLAGS.load
     49             name = loader.model_name(FLAGS.load)
AssertionError: bin/yolo.weights not found 

我尝试将bin和cfg文件夹从提取darkflow的原始目录复制到site packages/darkflow文件夹:

C:\Users\umair\AppData\Local\Enthought\Canopy\edm\envs\User\Lib\site-packages\darkflow

但这似乎也不管用。具有讽刺意味的是,在python控制台中,通过canopy命令提示符可以很好地工作 canopy command prompt

那么这些基本上有什么问题?我在安装软件包时犯了什么错误吗?错过了在某处添加路径?或者我把bin和cfg文件夹放在了错误的位置?在


Tags: pyselfmodelbinpackagessiteloadyolo
1条回答
网友
1楼 · 发布于 2024-10-02 18:19:05

听起来好像代码假设它与脚本运行在同一个目录中。这是不能假设的。您可以在caopy编辑器中的ipython提示符下更改工作目录,也可以配置Canopy来自动执行此操作。请参阅树冠文档:https://docs.enthought.com/canopy/2.1/quick-start/code_editor.html#change-directory

相关问题 更多 >