convnextv2-huge-upsample-exp1

ConvNeXt V2 Huge fine-tuned on CIFAR-10 upsampled to 224x224.

Fine-tuned from ImageNet-pretrained weights (timm name convnextv2_huge) on CIFAR-10 as part of a native-resolution vs upsampling study.

Evaluation results (CIFAR-10 test set, 10,000 images)

Metric Value
Top-1 accuracy 99.52%
Top-5 accuracy 99.97%
F1 (macro) 0.9952
AUC (macro) 0.9997
ECE 0.1060
  • Parameters: 657.50M
  • Input: CIFAR images upsampled to 224x224
  • Preprocessing: resize to 224x224, then normalize with CIFAR-10 mean [0.4914, 0.4822, 0.4465] and std [0.247, 0.2435, 0.2616].

Usage

from cifar_classifier import CIFAR10Classifier

clf = CIFAR10Classifier.from_pretrained("DeKUT-DSAIL/convnextv2-huge-cifar10-upsample")
label, probs = clf.predict("image.jpg")
print(label, probs)

Get cifar_classifier.py from the hf_deployment/ folder of the training repository. It only needs torch, torchvision, timm, Pillow and huggingface_hub.

Classes

airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck

Downloads last month
-
Safetensors
Model size
0.7B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train DeKUT-DSAIL/convnextv2-huge-cifar10-upsample