tyre

build.py

model= Sequential([
    layers.Conv2D(32, (3,3), activation= 'relu', input_shape= (100,100,1)),
    layers.MaxPooling2D(pool_size= (2,2), padding= 'same'),
    layers.Dropout(0.3),
    layers.Conv2D(16, (3,3), activation= 'relu'),
    layers.MaxPooling2D(pool_size= (2,2), padding= 'same'),
    layers.Flatten(),
    layers.Dropout(0.3),
    layers.Dense(25, activation= 'relu'),
#     layers.Dense(64, activation= 'relu'),
    layers.Dense(3, activation= 'softmax')
])
model.compile(optimizer= 'adam', loss= 'categorical_crossentropy', metrics= ['accuracy'])