BBox贴标工具

时间:2018-10-24 05:04:52

标签: python python-2.7 computer-vision yolo

我正在尝试在自定义图片上训练YOLOv2,并且正在关注此链接 https://timebutt.github.io/static/how-to-train-yolov2-to-detect-custom-objects/

为注释对象,我正在使用BBox Labeling Tool。上面的链接要求我更改BBox Labeling Tool文件夹中main.py文件中第126行的目录。更改行后,代码如下所示。

s = r'../DATA/001/'

运行代码并单击“加载图像”按钮时,我收到以下错误消息

File "C:\Users\user\Anaconda3\envs\yolo_training\lib\lib-tk\Tkinter.py", line 1541, in __call__
    return self.func(*args)
  File "main.py", line 127, in loadDir
    self.category = int(s)
ValueError: invalid literal for int() with base 10: ''

我用anaconda用python 2.7创建了一个虚拟环境,因为BBox标签工具需要python 2.7。

我不知道我在做什么错。请帮助我。

谢谢

PS:001文件夹中的所有图像均为jpeg文件,这也是BBox标签工具的要求

1 个答案:

答案 0 :(得分:1)

据我了解,这应该是工具本身的路径。

s = r'D:\workspace\python\labelGUI'

我看不到说明中要求您更改此路径的位置。这与我认为的调试有关。我建议您还原路径。

在这种情况下,您只需要在工具目录中的图像文件夹内创建一个名为“ 00X”(X应该是数字)的文件夹,就像示例中所示。创建文件夹后,假设您将“ 005”放在其中。 加载之前,请选择数字5以指定您的文件夹。