gt_argmax_overlaps=overlaps.argmax(axis=0) ValueError: attempt to get argmax of an empty sequence

问题描述

用 Faster-RCNN 训练检测模型时，训练到中间过程报如下错误：

gt_argmax_overlaps = overlaps.argmax(axis=0)  
ValueError: attempt to get argmax of an empty sequence

解决方法

写一段代码检查数据问题，可能的问题包括：
1、生成了 xml 标注文件，但标注数量为空
2、检测 xml 中的 width、height 是否为 0
3、检测 size 的 xmin、xmax、ymin、ymax 是否在 [0, width]、[0, height] 范围内

输出异常 xml 路径，将其删除或者修正即可再次训练模型了，训练一切正常。

• 注：我训练模型用的数据是VOC格式