Muselion/Test_Dogs

🖼️ ConvNeXt-Tiny Custom Classifier

Ce modèle est basé sur facebook/convnext-tiny-224, avec une tête de classification adaptée à une tâche personnalisée de classification d'images.

🧠 Architecture

• Modèle de base : ConvNeXt-Tiny
• Nombre de classes : 𝑁 (ex. 5)
• Tête personnalisée : torch.nn.Linear(..., out_features=N)

🏗️ Prétraitement des images

Le modèle n'utilise pas de AutoImageProcessor. Vous devez appliquer manuellement les transformations suivantes avant d'inférer :

from torchvision import transforms

val_transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(
        mean=[0.485, 0.456, 0.406], 
        std=[0.229, 0.224, 0.225]
    )
])