Faster R-CNN 中的锚点大小调整答案

【问题标题】：Anchor size tuning in Faster R-CNNFaster R-CNN 中的锚点大小调整
【发布时间】：2021-08-26 17:37:30
【问题描述】：

我正在训练 Faster R-CNN 从无人机摄影中检测小物体：25x25 到 30x30 像素。到目前为止，我已经使用 .config 文件中给出的默认锚点尺寸完成了这项工作：

first_stage_anchor_generator {
      grid_anchor_generator {
        height_stride: 16
        width_stride: 16
        scales: 0.25
        scales: 0.5
        scales: 1.0
        scales: 2.0
        aspect_ratios: 0.5
        aspect_ratios: 1.0
        aspect_ratios: 2.0

但是，我在文献中读到以下内容：

原始的 Faster-RCNN 模型参数是在使用前视图数据集获取的数据集上进行训练的，因此我们通过将锚点的大小减小到（8、16、32、64、128）来重新训练模型以适应俯视无人机图像更好的。网络及其修改的这种组合显着提高了模型在我们数据集上的准确性和召回率。

我已经通过网站阅读了很多内容，但仍然无法确定我应该修改哪些参数才能将锚大小调整为文献建议的大小（8、16、32、64、128）。

任何帮助将不胜感激！

【问题讨论】：

配置文件使用什么代码库？是在 github 上还是可以提供链接或代码？
嗨 @TCArlen 这里是 github repo 的 link，包含代码库的文件名为 grid_anchor_generator.py

标签： python tensorflow hyperparameters faster-rcnn anchorpoint

【解决方案1】：

您可以在 grid_anchor_generator.proto 中找到网格锚生成器的默认值。如下图所示

message GridAnchorGenerator {
   // Anchor height in pixels.
  optional int32 height = 1 [default = 256];

  // Anchor width in pixels.
  optional int32 width = 2 [default = 256];

我认为，将以下参数添加到更快的 r-cnn 配置文件将解决您的问题。

first_stage_anchor_generator {
      grid_anchor_generator {
        scales: [1.0, 2.0 , 3.0 , 4.0 , 5.0 , 6.0 ]
        aspect_ratios: [0.5, 1.0, 2.0]
        height_stride: 16
        width_stride: 16
        height: 8 # or you can change any value you desired (default = 256)
        width: 8 # or you can change any value you desired (default = 256)
      }
    }

【讨论】：