CANet-v1 / README.md
CernovaAI's picture
Update README.md
05ab0ba verified
metadata
license: mit

🧠 Multi-Cancer Image Classification

This project aims to classify different types of cancer images using a deep learning Convolutional Neural Network (CNN). The model is built with TensorFlow/Keras.

πŸ“‚ Project Structure

  • Data preprocessing is done with ImageDataGenerator (normalization and train/validation split).
  • The CNN model is created with Conv2D, MaxPooling2D, and Dense layers.
  • After training, the model is saved as .h5.
  • The function gercek_deger() allows testing predictions on a single image.

βš™οΈ Dependencies

  • TensorFlow / Keras
  • NumPy
  • Matplotlib

πŸš€ Model Architecture

  1. Conv2D + MaxPooling2D β†’ Feature extraction from images
  2. Conv2D + MaxPooling2D
  3. Conv2D + MaxPooling2D
  4. Flatten β†’ Convert data into a vector
  5. Dense (512, ReLU) β†’ Fully connected layer
  6. Dense (Softmax) β†’ Output layer with class probabilities

πŸ‹οΈ Training

  • Input image size: 150x150
  • Batch size: 32
  • Optimizer: Adam
  • Loss: Categorical Crossentropy
  • Metrics: Accuracy
  • Epochs: 10

πŸ“Š Usage

Train the model

model.fit(train_generator, validation_data=validation_generator, epochs=10)

Save the model

model.save("image_classifier.h5")

Predict on a new image

gercek_deger("test_image.jpg", model, train_generator.class_indices)