NameError:未定义名称'exit‘/tfnet/暗流/自定义数据集对象检测

Question

我试图通过使用yolov3.cfg创建带有标签注释和图像的yolov3.weights文件和darkflow来训练我的自定义数据集。然而，当我试图运行tfnet = TFNet(history)时，它会抛出"exit not“的错误。

我已经通过以下步骤安装了darkflow：

在Anaconda提示符：

git clone https://github.com/thtrieu/darkflow.git
cd darkflow
python3 setup.py build_ext –inplace
pip install

然后：

%matplotlib inline

import matplotlib.pyplot as plt

import numpy as np


import cv2

from darkflow.net.build import TFNet

history = {"model": "C:/Users/Business Intelli/Desktop/Object-Detection/Dataset/yolov3.cfg",

           "load": "C:/Users/Business Intelli/Desktop/Object-Detection/Dataset/yolov3.weights",

           "batch": 8,

           "epoch": 50,

           "gpu": 1.0,

           "train": True,

           "annotation": "C:/Users/Business Intelli/Desktop/Object-Detection/Dataser/Stumps",

           "dataset": "C:/Users/Business Intelli/Desktop/Object-Detection/Dataser/Stumps"}


tfnet = TFNet(history)

Parsing C:/Users/Business Intelli/Desktop/Object-Detection/Dataset/yolov3.cfg
---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-10-6f6b945047c5> in <module>
----> 1 tfnet = TFNet(history)

~\Anaconda3\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 

~\Anaconda3\lib\site-packages\darkflow\dark\darknet.py in __init__(self, FLAGS)

     15 

     16         print('Parsing {}'.format(self.src_cfg))

---> 17         src_parsed = self.parse_cfg(self.src_cfg, FLAGS)

     18         self.src_meta, self.src_layers = src_parsed

     19 


~\Anaconda3\lib\site-packages\darkflow\dark\darknet.py in parse_cfg(self, model, FLAGS)

     66         cfg_layers = cfg_yielder(*args)

     67         meta = dict(); layers = list()

---> 68         for i, info in enumerate(cfg_layers):

     69             if i == 0: meta = info; continue

     70             else: new = create_darkop(*info)


~\Anaconda3\lib\site-packages\darkflow\utils\process.py in cfg_yielder(model, binary)

    314                 #-----------------------------------------------------

    315                 else:

--> 316                         exit('Layer {} not 
implemented'.format(d['type']))

    317 

    318                 d['_size'] = list([h, w, c, l, flat])


NameError: name 'exit' is not defined

Answer 1

因此，我也遇到了同样的问题，问题是process.py文件在黑暗流-> utils文件夹中。

显然，exit()方法不是内置的，所以您必须在process.py中添加这一行。

从sys导入退出的

注意:如果您的代码达到了这一点，这意味着模型无法读取这些层。我为yolov3下载的权重文件给我带来了同样的麻烦，我找不到一个在黑暗流中为yolov3提供适当权重文件的链接。所以我不得不坚持yolo.cfg和yolo.weights。