Calculateur de taille de sortie de convolution CNN
Calculez la taille spatiale de sortie, le champ récepteur et les FLOPs par pixel d'une convolution 2D à partir de la taille d'entrée, du noyau, du pas et du rembourrage.
Comment utiliser le calculateur de taille de sortie de convolution CNN
- Entrez la taille spatiale d'entrée.
- Entrez la taille du noyau, le pas et le rembourrage.
- Cliquez sur Calculer pour voir la taille de sortie et le champ récepteur.
Cas d'utilisation
- •Dimensionner manuellement un CNN pour un appareil embarqué.
- •Déboguer les incompatibilités de forme dans PyTorch.
- •Enseigner les fondamentaux de la convolution.
Formule
Sortie = ⌊(entrée + 2 × rembourrage − noyau) / pas⌋ + 1. Le champ récepteur pour une couche est égal à la taille du noyau.
Questions fréquemment posées
Qu'est-ce que le rembourrage "same" ?
Un rembourrage qui maintient la taille de sortie égale à la taille d'entrée. Pour un noyau k et un pas de 1, le rembourrage requis est ⌊k/2⌋.
Pourquoi la sortie doit-elle être un entier ?
Une sortie fractionnaire indique un rembourrage mal ajusté. La plupart des frameworks appliquent un arrondi par défaut au résultat, perdant la rangée de pixels droite ou inférieure — c'est pourquoi un rembourrage asymétrique est parfois utilisé.
Cela inclut-il le biais et les canaux ?
Non. Multipliez les paramètres et les FLOPs par les canaux d'entrée × de sortie pour obtenir les totaux par couche, et ajoutez un terme de biais par canal de sortie.