实现Faster R-CNN对象检测算法时出现错误

Question

我正在尝试实现更快的R-CNN对象检测算法，我有一个不寻常的错误。在尝试调用这个协作教程中的train_one_epoch函数时，我在这里提到的loss_dict = model(images, targets)中出现了错误。确切的错误是:

101         cell_anchors = self.cell_anchors
102         assert cell_anchors is not None
--> 103         assert len(grid_sizes) == len(strides) == len(cell_anchors)
104 
105         for size, stride, base_anchors in zip(
AssertionError:

有人知道吗?提前感谢!

Answer 1

最后，我能够解决这个问题，它只是通过在Faster R-CNN函数中添加调整AnchorGenerator的大小及其相应的长宽比

ft_anchor_generator = AnchorGenerator(
sizes=((32, 64, 128),), aspect_ratios=((0.5, 1.0, 2.0),)
)
ft_model = FasterRCNN(
backbone=ft_backbone,
num_classes=num_classes,
rpn_anchor_generator=ft_anchor_generator)