Muselion commited on
Commit
96306e4
·
verified ·
1 Parent(s): 6a16c5a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -3
README.md CHANGED
@@ -1,3 +1,27 @@
1
- ---
2
- license: unknown
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 🖼️ ConvNeXt-Tiny Custom Classifier
2
+
3
+ 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.
4
+
5
+ ## 🧠 Architecture
6
+
7
+ - Modèle de base : ConvNeXt-Tiny
8
+ - Nombre de classes : 𝑁 (ex. 5)
9
+ - Tête personnalisée : `torch.nn.Linear(..., out_features=N)`
10
+
11
+ ## 🏗️ Prétraitement des images
12
+
13
+ Le modèle **n'utilise pas de `AutoImageProcessor`**.
14
+ Vous devez appliquer manuellement les transformations suivantes avant d'inférer :
15
+
16
+ ```python
17
+ from torchvision import transforms
18
+
19
+ val_transform = transforms.Compose([
20
+ transforms.Resize(256),
21
+ transforms.CenterCrop(224),
22
+ transforms.ToTensor(),
23
+ transforms.Normalize(
24
+ mean=[0.485, 0.456, 0.406],
25
+ std=[0.229, 0.224, 0.225]
26
+ )
27
+ ])