labelImg ZeroDivisionError: float division by zero解决办法

问题

做目标检测数据集的时候，需要用到目标检测标记工具。目前使用的是labelImg,该项目地址：
https://github.com/tzutalin/labelImg
使用方法在网上搜索有一大堆，这里不做过多的介绍。
使用页面如下所示：

在使用的过程中，框选相关部分后，程序闪退，然后终端提示错误信息如下：

xcen = float((xmin + xmax)) / 2 / self.imgSize[1]
ZeroDivisionError: float division by zero

网上查询相关的issue，有人提出同样的问题，下面有相应的回答。

• https://github.com/tzutalin/labelImg/issues/386
• https://github.com/tzutalin/labelImg/issues/462

解决方法

问题出现的原因是该软件在打开某些图片时候，背景为透明的那种，会出现size为0的情况，解决办法是用opencv打开后，再重新保存覆盖原文件即可，代码如下：

import sys
import cv2

input_file = sys.argv[1]  # 支持批量操作
with open(input_file, 'r') as f:
    for line in f:
        img_name = line.rstrip('\n')
        img = cv2.imread(img_name)
        save_path = img_name
        cv2.imwrite(save_path, img)  # 可以加quality参数

上述代码支持批量操作，直接将整个文件夹下的文件名输入进去即可。进行覆盖保存，再次打开软件处理这些图片，不会报这一个错误