अगर आप AI और Computer Vision में शुरुआत करना चाहते हैं, तो Image Classification एक शानदार प्रोजेक्ट है। इस प्रोजेक्ट में हम Python और TensorFlow का उपयोग करके एक ऐसा मॉडल बनाएंगे जो तस्वीरों को Categories (जैसे Cat, Dog) में पहचान सके।
Bonus: Accuracy और Loss को Graph में Visualize करें
from tensorflow.keras.datasets import cifar10
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
# Load Data
(x_train, y_train), (x_test, y_test) = cifar10.load_data()
# Normalize
x_train, x_test = x_train / 255.0, x_test / 255.0
# Model
model = Sequential([
Conv2D(32, (3,3), activation='relu', input_shape=(32,32,3)),
MaxPooling2D((2,2)),
Flatten(),
Dense(64, activation='relu'),
Dense(10, activation='softmax')
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5, validation_data=(x_test, y_test))
