Add updated model card (README.md) with augmentation details
Browse files
README.md
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Model Card for augmented-donut-pose
|
| 2 |
+
|
| 3 |
+
## Model Details
|
| 4 |
+
* **Model Architecture**: resnet18
|
| 5 |
+
* **Pre-trained**: Yes
|
| 6 |
+
* **Number of Classes**: 2
|
| 7 |
+
|
| 8 |
+
## Dataset Details
|
| 9 |
+
* **Dataset Name**: lay-or-stand-donuts-dataset
|
| 10 |
+
* **Training Samples**: 246
|
| 11 |
+
* **Validation Samples**: 42
|
| 12 |
+
* **Test Samples**: 72
|
| 13 |
+
|
| 14 |
+
## Training Metrics
|
| 15 |
+
* **Final Training Loss**: 0.1101
|
| 16 |
+
* **Validation Accuracy**: 100.00%
|
| 17 |
+
|
| 18 |
+
## Data Augmentations
|
| 19 |
+
This model was trained with advanced data augmentations including `RandomResizedCrop`, `RandomHorizontalFlip`, `RandomRotation`, `ColorJitter`, and `RandomErasing` to improve generalization.
|
| 20 |
+
|
| 21 |
+
## Augmented Sample Images
|
| 22 |
+
This repository also contains a selection of augmented sample images generated during the data preparation phase, which can be found in the `augmented_samples/` directory.
|
| 23 |
+
|
| 24 |
+
## Usage
|
| 25 |
+
This model is a classification model trained to distinguish between two classes related to donuts (e.g., 'lay' or 'stand'). It was fine-tuned on the `lay-or-stand-donuts-dataset` dataset with enhanced data augmentations.
|
| 26 |
+
|