我正在尝试在自定义图像上训练 YOLOv2,并正在关注此链接 https://timebutt.github.io/static/how-to-train-yolov2-to-detect-custom-objects/
为了注释对象,我使用了BBox 标签工具。上面的链接要求我更改 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-labeling工具的要求
相关分类