Skip to content

Commit

Permalink
Update rcnn.md
Browse files Browse the repository at this point in the history
  • Loading branch information
astonzhang authored Mar 13, 2019
1 parent 72a6644 commit a3abeff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chapter_computer-vision/rcnn.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ R-CNN的主要性能瓶颈在于需要对每个提议区域独立抽取特征。
它的主要计算步骤如下。

1. 与R-CNN相比,Fast R-CNN用来提取特征的卷积神经网络的输入是整个图像,而不是各个提议区域。而且,这个网络通常会参与训练,即更新模型参数。设输入为一张图像,将卷积神经网络的输出的形状记为$1 \times c \times h_1 \times w_1$。
1. 假设选择性搜索生成$n$个提议区域。这些形状各异的提议区域在卷积神经网络的输出上分别标出形状各异的兴趣区域。这些兴趣区域需要抽取出形状相同的特征(假设高和宽均分别指定为$h_2$和$w_2$)以便于连结后输出。Fast R-CNN引入兴趣区域池化(Region of Interest Pooling,RoI池化)层,将卷积神经网络的输出和提议区域作为输入,输出连结后的各个提议区域抽取的特征,形状为$n \times c \times h_2 \times w_2$。
1. 假设选择性搜索生成$n$个提议区域。这些形状各异的提议区域在卷积神经网络的输出上分别标出形状各异的兴趣区域。这些兴趣区域需要抽取出形状相同的特征(假设高和宽均分别指定为$h_2$和$w_2$)以便于连结后输出。Fast R-CNN引入兴趣区域池化(region of interest pooling,RoI池化)层,将卷积神经网络的输出和提议区域作为输入,输出连结后的各个提议区域抽取的特征,形状为$n \times c \times h_2 \times w_2$。
1. 通过全连接层将输出形状变换为$n \times d$,其中超参数$d$取决于模型设计。
1. 预测类别时,将全连接层的输出的形状再变换为$n \times q$并使用softmax回归($q$为类别个数)。预测边界框时,将全连接层的输出的形状变换为$n \times 4$。也就是说,我们为每个提议区域预测类别和边界框。

Expand Down

0 comments on commit a3abeff

Please sign in to comment.