Question

按照https://github.com/deboc/py-faster-rcnn/blob/master/help/Readme.md

上的说明在自定义数据集上训练py-faster-rcnn

我遇到了一些错误，比如

AttributeError: 'numpy.ndarray' object has no attribute 'toarray' in py-faster-rcnn

我设法绕过编辑https://github.com/rbgirshick/py-faster-rcnn/blob/master/lib/roi_data_layer/roidb.py

gt_overlaps = roidb[i]['gt_overlaps']
gt_overlaps = sp.sparse.csr_matrix(gt_overlaps).toarray()

但是，在培训过程中，我收到了两次警告

RuntimeWarning: invalid value encountered in log targets_dw = np.log(gt_widths / ex_widths)

在文件https://github.com/rbgirshick/py-faster-rcnn/blob/master/lib/fast_rcnn/bbox_transform.py

中

结果会受此影响吗？

我需要做些不同的事吗？

Answer 1

也许你应该尝试修改“lib/datasets/pascal_voc.py”

函数“_load_pascal_annotation(,)”中的

正确的应该是：

x1 = float(bbox.find('xmin').text)                                                                 
y1 = float(bbox.find('ymin').text)                                      
x2 = float(bbox.find('xmax').text)                                                                 
y2 = float(bbox.find('ymax').text)

原因在于您自己的数据，x1或y1可能等于1，如果减1，则数字为负数，从而导致错误

在自定义数据集的py-faster-rcnn培训期间发出警告

1 个答案: