UD3 · Integración de modelos en aplicaciones | MP03
Redes Convolucionales (CNN): datos espaciales
import torch.nn as nn
class CNNClasificador(nn.Module):
"""
CNN para clasificacion de imagenes con arquitectura progressiva.
"""
def __init__(self, n_clases: int):
super().__init__()
self.extractor = nn.Sequential(
nn.Conv2d(3, 32, kernel_size=3, padding=1),
nn.BatchNorm2d(32), nn.ReLU(),
nn.MaxPool2d(2),
nn.Conv2d(32, 64, kernel_size=3, padding=1),
nn.BatchNorm2d(64), nn.ReLU(),
nn.MaxPool2d(2),
nn.Conv2d(64, 128, kernel_size=3, padding=1),
nn.BatchNorm2d(128), nn.ReLU(),
nn.AdaptiveAvgPool2d((7, 7))
)
self.clasificador = nn.Sequential(
nn.Flatten(),
nn.Linear(128 * 7 * 7, 512), nn.ReLU(), nn.Dropout(0.4),
nn.Linear(512, n_clases)
)
def forward(self, x):
return self.clasificador(self.extractor(x))