File size: 1,203 Bytes
0b5d0ca 05f2a72 0b5d0ca 05f2a72 0b5d0ca 271792f 30d9ac7 05f2a72 3e79bea 30d9ac7 271792f 91ba56c 1805169 2f96ba6 1805169 91ba56c 05f2a72 |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
---
tags:
- onnx
- image-classification
- cifar10
- dropout
- aidge
pipeline_tag: image-classification
datasets:
- cifar10
metrics:
- accuracy
model-index:
- name: Custom ResNet-18 with integrated Dropout layers
results:
- task:
type: image-classification
name: Image Classification
dataset:
name: CIFAR-10
type: cifar10
metrics:
- type: accuracy
value: 83.96%
language:
- en
base_model:
- resnet-18
---
# Custom ResNet-18 with integrated Dropout Layers
This is a **custom ResNet-18** ONNX model implemented in **PyTorch** with integrated **Dropout** layers.
It was trained on the **CIFAR-10** dataset for image classification tasks.
The model has been exported using **opset version 15** and is fully compatible with the **Aidge** platform.
## Details
- **Architecture**: Customized ResNet-18 with integrated Dropout layers
- **Trained on**: CIFAR-10 (60,000 32x32 color images, 10 classes)
- **Image Preprocessing**: Images were resized to `128×128`
- **Data Normalization**: `mean = [0.4914, 0.4822, 0.4465]` ; `std = [0.2023, 0.1994, 0.2010]`
- **Dropout Probability**: 0.3
- **ONNX opset version**: 15
- **Conversion tool**: PyTorch → ONNX
--- |