Downloads · 30 days
9
41% of all-time downloads
kentaromiz/augmented-donut-pose
augmented-donut-pose is a machine learning model from kentaromiz. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Model Architecture: resnet18 Pre-trained: Yes Number of Classes: 2
Downloads · 30 days
9
41% of all-time downloads
All-time downloads
22
Public
Repo size
89.6 MB
Likes
0
Public
Click a slice to open those files.
.pth44.8 MB · 99%
From the Hugging Face model README
This model was trained with advanced data augmentations including RandomResizedCrop, RandomHorizontalFlip, RandomRotation, ColorJitter, and RandomErasing to improve generalization.
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.
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.