Commit
•
d5f4402
1
Parent(s):
7893505
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,8 +11,6 @@ Predicts clothing features.
|
|
| 11 |
|
| 12 |
## How to use
|
| 13 |
|
| 14 |
-
https://huggingface.co/learn/computer-vision-course/en/unit3/vision-transformers/vision-transformers-for-image-classification
|
| 15 |
-
|
| 16 |
```python
|
| 17 |
import torch
|
| 18 |
import torchvision.transforms as transforms
|
|
@@ -68,4 +66,7 @@ def predict_features(image_path):
|
|
| 68 |
return pred_labels
|
| 69 |
|
| 70 |
print(predict_features('image.jpg'))
|
| 71 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
## How to use
|
| 13 |
|
|
|
|
|
|
|
| 14 |
```python
|
| 15 |
import torch
|
| 16 |
import torchvision.transforms as transforms
|
|
|
|
| 66 |
return pred_labels
|
| 67 |
|
| 68 |
print(predict_features('image.jpg'))
|
| 69 |
+
```
|
| 70 |
+
|
| 71 |
+
## References
|
| 72 |
+
https://huggingface.co/learn/computer-vision-course/en/unit3/vision-transformers/vision-transformers-for-image-classification
|