Fast R-CNN

O Fast R-CNN é a evolução do r-cnn, criado para resolver o gargalo extremo de lentidão ao processar imagens, introduzindo o compartilhamento de computação convolucional e a técnica de RoI Pooling.

Como o Fast R-CNN resolveu os problemas do R-CNN

No R-CNN original, a rede passava independentemente por ~2.000 regiões propostas, o que era muito redundante e lento. O Fast R-CNN mudou esse paradigma:

1. Compartilhamento de Features: Em vez de recortar a imagem 2.000 vezes, a imagem inteira é passada pela CNN uma única vez para produzir um único feature map gigante.

2. Projeção da Proposta: O algoritmo de Selective Search ainda propõe as ~2.000 regiões, mas agora essas coordenadas são projetadas diretamente em cima do feature map já calculado, e não na imagem original.

3. RoI Pooling: Para cada região no feature map (que tem tamanhos variados), aplica-se a camada de Region of Interest Pooling, que converte a região para um tamanho fixo (essencial para as camadas densas finais). Nota: O RoI Pooling seria depois melhorado pelo roi-align no Mask R-CNN devido a problemas de arredondamento.

4. Rede Multi-task: As SVMs lentas foram substituídas por uma camada Softmax, permitindo que a classificação e o refinamento do bounding box fossem treinados juntos na mesma rede. Isso é feito otimizando uma função de perda (Loss) conjunta (Multi-task Loss):

   $$L(p, u, t^u, v) = L_{cls}(p, u) + \lambda [u \ge 1] L_{loc}(t^u, v)$$

   Eq. 1: Função de perda multitarefa unificando classificação e regressão.

   Onde:

   • $p$: distribuição de probabilidade calculada pela Softmax sobre as categorias da rede.
   • $u$: rótulo real (ground truth) da classe ($u=0$ representa fundo/background).
   • $t^u$: offsets (ajustes) preditos pelo regressor de bounding box para a classe $u$.
   • $v$: offsets reais (ground truth) da bounding box.
   • $\lambda$: hiperparâmetro de balanceamento entre classificação e localização.
   • $[u \ge 1]$: função indicadora que ativa a perda de regressão apenas se a proposta for um objeto real (ignorando fundos).

Vantagens e a Nova Limitação

• Treinamento End-to-End: Classificador e regressor agora são treinados simultaneamente.
• Muito mais rápido: Cerca de 200x mais rápido que o R-CNN clássico em tempo de inferência (processando uma imagem em ~2 segundos).

⚠️ O Gargalo Remanescente: A única coisa que ainda impedia o Fast R-CNN de rodar em tempo real era a dependência do Selective Search para propor regiões (que rodava na CPU). Esse problema foi resolvido posteriormente pelo faster-r-cnn.

Relacionadas: r-cnn · faster-r-cnn · roi-align