Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+cat_dog_classifier.keras filter=lfs diff=lfs merge=lfs -text

cat_dog_classifier.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a199caa07ed268b554abeba28751f69eedb73ba58dc1ee4984bb320dab1c1d3
+size 59234141

predictor.py

@@ -0,0 +1,32 @@
+import cv2  # Assuming you have OpenCV installed
+import numpy as np
+from tensorflow.keras.preprocessing import image
+import tensorflow as tf
+# Load the saved model
+model = tf.keras.models.load_model('cat_dog_classifier.keras')  # Replace with your model filename
+img_width, img_height = 224, 224  # VGG16 expects these dimensions
+
+
+# Function to preprocess an image for prediction
+def preprocess_image(img_path):
+    img = cv2.imread(img_path)  # Read the image
+    img = cv2.resize(img, (img_width, img_height))  # Resize according to model input size
+    img = img.astype('float32') / 255.0  # Normalize pixel values
+    img = np.expand_dims(img, axis=0)  # Add a batch dimension (model expects batch of images)
+    return img
+
+
+# Get the path to your new image
+new_image_path = 'test1/11.jpg'  # Replace with your image path
+
+# Preprocess the image
+preprocessed_image = preprocess_image(new_image_path)
+
+# Make prediction
+prediction = model.predict(preprocessed_image)
+
+# Decode the prediction (assuming class 0 is cat, 1 is dog)
+predicted_class = int(prediction[0][0] > 0.5)  # Threshold of 0.5 for binary classification
+class_names = ['cat', 'dog']  # Adjust class names according to your model
+
+print(f"Predicted class: {class_names[predicted_class]}")

train.py

@@ -0,0 +1,78 @@
+import tensorflow as tf
+from tensorflow.keras.preprocessing.image import ImageDataGenerator
+from tensorflow.keras.applications import VGG16
+from tensorflow.keras.layers import Flatten, Dense
+
+# Define data paths (modify as needed)
+train_data_dir = 'tt'
+validation_data_dir = 'valid'
+test_data_dir = 'valid'
+
+# Set image dimensions (adjust if necessary)
+img_width, img_height = 224, 224  # VGG16 expects these dimensions
+
+# Data augmentation for improved generalization (optional)
+train_datagen = ImageDataGenerator(
+    rescale=1./255,  # Normalize pixel values
+    shear_range=0.2,
+    zoom_range=0.2,
+    horizontal_flip=True,
+    fill_mode='nearest'
+)
+
+validation_datagen = ImageDataGenerator(rescale=1./255)  # Only rescale for validation
+
+# Load training and validation data
+train_generator = train_datagen.flow_from_directory(
+    train_data_dir,
+    target_size=(img_width, img_height),
+    batch_size=32,  # Adjust batch size based on GPU memory
+    class_mode='binary'  # Two classes: cat or dog
+)
+
+validation_generator = validation_datagen.flow_from_directory(
+    validation_data_dir,
+    target_size=(img_width, img_height),
+    batch_size=32,
+    class_mode='binary'
+)
+
+# Load pre-trained VGG16 model (without the top layers)
+base_model = VGG16(weights='imagenet', include_top=False, input_shape=(img_width, img_height, 3))
+
+# Freeze the base model layers (optional - experiment with unfreezing for fine-tuning)
+base_model.trainable = False
+
+# Add custom layers for classification on top of the pre-trained model
+x = base_model.output
+x = Flatten()(x)
+predictions = Dense(1, activation='sigmoid')(x)  # Sigmoid for binary classification
+
+# Create the final model
+model = tf.keras.Model(inputs=base_model.input, outputs=predictions)
+
+# Compile the model
+model.compile(loss='binary_crossentropy',
+              optimizer='adam',
+              metrics=['accuracy'])
+
+# Train the model
+history = model.fit(
+    train_generator,
+    epochs=3,  # Adjust number of epochs based on dataset size and validation performance
+    validation_data=validation_generator
+)
+
+# Evaluate the model on test data (optional)
+test_generator = validation_datagen.flow_from_directory(
+    test_data_dir,
+    target_size=(img_width, img_height),
+    batch_size=32,
+    class_mode='binary'
+)
+
+test_loss, test_acc = model.evaluate(test_generator)
+print('Test accuracy:', test_acc)
+
+# Save the model for future use (optional)
+model.save('cat_dog_classifier2.keras')