File size: 858 Bytes
96306e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# 🖼️ ConvNeXt-Tiny Custom Classifier
Ce modèle est basé sur [`facebook/convnext-tiny-224`](https://huggingface.co/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 :
```python
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]
)
]) |