Hugging Face's logo

Raditsoic
/
trashnet

Image Classification
Keras
Model card Files
trashnet
File size: 472 Bytes 
655e560
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
{
    "architecture": "Custom CNN",
    "input_size": [
        224,
        224
    ],
    "num_classes": 6,
    "transformations": [
        "RandomHorizontalFlip",
        "RandomRotation(15)",
        "ColorJitter(brightness=0.2, contrast=0.2)",
        "Resize(224x224)",
        "ToTensor",
        "Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])"
    ],
    "optimizer": "Adam",
    "learning_rate": 0.001,
    "batch_size": 32,
    "epochs": 10
}