Hugging Face's logo

hlopez
/
ViT_waste_classifier

English
Model card Files
xet
 Community
hlopez commited on
Commit
7cc5a74
·
1 Parent(s): efef84f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -12,3 +12,6 @@ This model is a ViT trained to classify waste images into 6 categories:
12
 
13
  The repository related to this model is: https://github.com/hectorLop/Waste-Detector
14
  Also, the code related to this model can be found here https://github.com/hectorLop/Waste-Detector/blob/main/waste_detector/classifier/sagemaker/model.py
 
 
 
 
12
 
13
  The repository related to this model is: https://github.com/hectorLop/Waste-Detector
14
  Also, the code related to this model can be found here https://github.com/hectorLop/Waste-Detector/blob/main/waste_detector/classifier/sagemaker/model.py
15
+
16
+ ### Requirements
17
+ - Works with RGB images of size 224x224