Instructions to use litert-community/squeezenet1_0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LiteRT
How to use litert-community/squeezenet1_0 with LiteRT:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
- Notebooks
- Google Colab
- Kaggle
Merge branch 'pr/2'
Browse files
README.md
CHANGED
|
@@ -26,13 +26,10 @@ model-index:
|
|
| 26 |
value: 0.8044
|
| 27 |
---
|
| 28 |
|
| 29 |
-
#
|
| 30 |
|
| 31 |
SqueezeNet 1.0 model pre-trained on ImageNet-1k at resolution 224x224. It was introduced in [SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size](https://arxiv.org/abs/1602.07360) by Forrest N. Iandola, Song Han, Matthew W. Moskewicz, Khalid Ashraf, William J. Dally, and Kurt Keutzer.
|
| 32 |
|
| 33 |
-
## Intended uses & limitations
|
| 34 |
-
|
| 35 |
-
The model files were converted from pretrained weights from PyTorch Vision. The models may have their own licenses or terms and conditions derived from PyTorch Vision and the dataset used for training. It is your responsibility to determine whether you have permission to use the models for your use case.
|
| 36 |
|
| 37 |
## Model description
|
| 38 |
|
|
@@ -43,9 +40,34 @@ acc@1 (on ImageNet-1K): 58.092%
|
|
| 43 |
acc@5 (on ImageNet-1K): 80.420%
|
| 44 |
num_params: 1,248,424
|
| 45 |
|
| 46 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
-
---
|
| 49 |
```python
|
| 50 |
#!/usr/bin/env python3
|
| 51 |
import argparse
|
|
@@ -117,8 +139,18 @@ def main():
|
|
| 117 |
|
| 118 |
if __name__ == "__main__":
|
| 119 |
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
|
| 121 |
-
### BibTeX Entry and Citation Info
|
| 122 |
```bibtex
|
| 123 |
@misc{iandola2016squeezenetalexnetlevelaccuracy50x,
|
| 124 |
title={SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size},
|
|
|
|
| 26 |
value: 0.8044
|
| 27 |
---
|
| 28 |
|
| 29 |
+
# Squeezenet1_0
|
| 30 |
|
| 31 |
SqueezeNet 1.0 model pre-trained on ImageNet-1k at resolution 224x224. It was introduced in [SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size](https://arxiv.org/abs/1602.07360) by Forrest N. Iandola, Song Han, Matthew W. Moskewicz, Khalid Ashraf, William J. Dally, and Kurt Keutzer.
|
| 32 |
|
|
|
|
|
|
|
|
|
|
| 33 |
|
| 34 |
## Model description
|
| 35 |
|
|
|
|
| 40 |
acc@5 (on ImageNet-1K): 80.420%
|
| 41 |
num_params: 1,248,424
|
| 42 |
|
| 43 |
+
This model is released under the BSD 3-Clause License, inheriting the license of the `torchvision` repository from which it was converted.
|
| 44 |
+
|
| 45 |
+
|
| 46 |
+
## Intended uses & limitations
|
| 47 |
+
|
| 48 |
+
The model files were converted from pretrained weights from PyTorch Vision. The models may have their own licenses or terms and conditions derived from PyTorch Vision and the dataset used for training. It is your responsibility to determine whether you have permission to use the models for your use case.
|
| 49 |
+
|
| 50 |
+
The preprocessing script below has been adjusted to handle standard ImageNet resize (256) and central crop (224) requirements. It explicitly strips away the legacy PyTorch `(B, C, H, W)` layout and adds the required Batch dimension to match the LiteRT **`(B, H, W, C)`** (NHWC) runtime expectation.
|
| 51 |
+
|
| 52 |
+
## How to Use
|
| 53 |
+
|
| 54 |
+
**1. Install Dependencies**
|
| 55 |
+
|
| 56 |
+
Ensure your Python environment is set up with the required libraries. Run the following command in your terminal
|
| 57 |
+
|
| 58 |
+
```bash
|
| 59 |
+
pip install numpy Pillow huggingface_hub ai-edge-litert
|
| 60 |
+
```
|
| 61 |
+
|
| 62 |
+
**2. Prepare Your Image**
|
| 63 |
+
|
| 64 |
+
The script expects an image file to analyze. Make sure you have an image (e.g., cat.jpg or car.png) saved in the same working directory as your script.
|
| 65 |
+
|
| 66 |
+
|
| 67 |
+
**3. Save the Script**
|
| 68 |
+
|
| 69 |
+
Create a new file named `classify.py`, paste the script below into it, and save the file:
|
| 70 |
|
|
|
|
| 71 |
```python
|
| 72 |
#!/usr/bin/env python3
|
| 73 |
import argparse
|
|
|
|
| 139 |
|
| 140 |
if __name__ == "__main__":
|
| 141 |
main()
|
| 142 |
+
```
|
| 143 |
+
|
| 144 |
+
**4. Execute the Python Script**
|
| 145 |
+
|
| 146 |
+
Run the below command:
|
| 147 |
+
|
| 148 |
+
```bash
|
| 149 |
+
python classify.py --image cat.jpg
|
| 150 |
+
```
|
| 151 |
+
|
| 152 |
+
### BibTeX entry and citation info
|
| 153 |
|
|
|
|
| 154 |
```bibtex
|
| 155 |
@misc{iandola2016squeezenetalexnetlevelaccuracy50x,
|
| 156 |
title={SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size},
|