Spaces:
Build error
Build error
Commit
·
6114c6c
1
Parent(s):
b555f64
Update Readme
Browse files
README.md
CHANGED
|
@@ -12,3 +12,18 @@ short_description: ResNet 50 model trained on Imagenet-1k data.
|
|
| 12 |
---
|
| 13 |
|
| 14 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
| 15 |
+
|
| 16 |
+
# ImageNet-1K Classification Model
|
| 17 |
+
|
| 18 |
+
This is a deep learning model trained on the ImageNet-1K dataset, capable of classifying images into 1000 different categories.
|
| 19 |
+
|
| 20 |
+
## Usage
|
| 21 |
+
|
| 22 |
+
Upload an image using the interface and the model will return the top 5 most likely classifications along with their probabilities.
|
| 23 |
+
|
| 24 |
+
## Model Details
|
| 25 |
+
|
| 26 |
+
- Architecture: ResNet-based model trained using PyTorch Lightning
|
| 27 |
+
- Dataset: ImageNet-1K
|
| 28 |
+
- Input: Images (recommended size: 224x224)
|
| 29 |
+
- Output: Probabilities for 1000 ImageNet classes
|