EnginCN55 commited on
Commit
ef06514
·
verified ·
1 Parent(s): d53b30e

Create README.md

Browse files

Brain Tumor Detection Model Training
/

Beyin Tümörü Tespiti Model Eğitimi
Proje Açıklaması / Project Description
Bu proje, beyin tümörü MRI görüntülerini sınıflandırmak için çeşitli derin öğrenme modellerini kullanarak transfer öğrenme ve özel CNN mimarisi ile eğitim yapmaktadır. Model eğitimleri sonunda TensorFlow Lite formatında çıktı alınmış ve eğitim süreçlerine ait performans grafikleri ile konfüzyon matrisleri oluşturulmuştur.
This project trains various deep learning models using transfer learning and custom CNN architecture to classify brain tumor MRI images. At the end of training, models are converted to TensorFlow Lite format, and training performance graphs along with confusion matrices are generated.



Kullanılan Modeller / Models Used
MobileNetV2 (Transfer Learning)

ResNet50 (Transfer Learning)




Eğitim Parametreleri / Training Parameters

| Parametre / Parameter | Değer / Value |
| ---------------------------- | ------------- |
| Görüntü Boyutu / Image Size | 224x224 |
| Batch Size | 32 |
| Epoch Sayısı / Epochs | 15 |
| Öğrenme Hızı / Learning Rate | 0.0001 (1e-4) |



Eğitim Süreci / Training Process
Veriler dataset/Training ve dataset/Testing klasörlerinden okunmuştur.

Data augmentation uygulanarak (döndürme, zoom, yatay ve dikey çevirme) eğitim seti çeşitlendirilmiştir.

Transfer öğrenme modellerinde son %20 katman serbest bırakılarak eğitim yapılmıştır.

EarlyStopping ve ReduceLROnPlateau callback’leri kullanılarak aşırı öğrenme önlenmiş ve öğrenme hızı dinamik olarak ayarlanmıştır.

Her model için eğitim kayıpları ve doğrulukları grafik olarak kaydedilmiştir.

Test verisinde F1 skorları ve sınıflandırma raporları hesaplanmıştır.

Eğitilen modeller TensorFlow Lite formatına dönüştürülüp tflite_models klasörüne kaydedilmiştir.



Kullanılan Kütüphaneler / Libraries Used
TensorFlow

NumPy

scikit-learn

Matplotlib















![MobileNetV2_confmat.png](https://cdn-uploads.huggingface.co/production/uploads/68501da68301dcec15632c63/wX_MXpjh9o_6E6mQqZYj8.png)
![MobileNetV2_history.png](https://cdn-uploads.huggingface.co/production/uploads/68501da68301dcec15632c63/vCFn2TwlVDnrdmynDcqIf.png)
![ResNet50_confmat.png](https://cdn-uploads.huggingface.co/production/uploads/68501da68301dcec15632c63/C_Vwwb3-Z0_w9RmEmPfDh.png)
![ResNet50_history.png](https://cdn-uploads.huggingface.co/production/uploads/68501da68301dcec15632c63/xweZUJwRkF-azX0SFa4Hm.png)

Files changed (1) hide show
  1. README.md +75 -0
README.md ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - tr
5
+ - en
6
+ metrics:
7
+ - accuracy
8
+ - f1
9
+ - confusion_matrix
10
+ base_model:
11
+ - microsoft/resnet-50
12
+ pipeline_tag: image-classification
13
+ tags:
14
+ - biology
15
+ ---
16
+ Model Summary
17
+ This model card describes two deep learning models trained to classify brain tumor MRI images into different tumor types. The models are based on ResNet50 and MobileNetV2 architectures and were trained using the Brain Tumor MRI Dataset available on Kaggle. They aim to assist medical professionals in detecting brain tumors using transfer learning approaches.
18
+ Model Details
19
+ ResNet50
20
+
21
+ - Developed by: [Your Name or Organization]
22
+ - Model type: Convolutional Neural Network (CNN)
23
+ - Language(s): N/A (Image classification task)
24
+ - License: [Specify license, e.g., MIT, Apache 2.0]
25
+ - Finetuned from model: ResNet50 (ImageNet pretrained)
26
+
27
+ MobileNetV2
28
+
29
+ - Developed by: [Your Name or Organization]
30
+ - Model type: Convolutional Neural Network (CNN)
31
+ - Language(s): N/A (Image classification task)
32
+ - License: [Specify license, e.g., MIT, Apache 2.0]
33
+ - Finetuned from model: MobileNetV2 (ImageNet pretrained)
34
+
35
+ Uses
36
+ Direct Use
37
+ These models can be used to classify MRI scans for brain tumor detection in clinical decision-support systems.
38
+ Out-of-Scope Use
39
+ Not intended for standalone diagnostic purposes without medical supervision. Misuse includes deployment without validation or interpretability assessments.
40
+ Bias, Risks, and Limitations
41
+ The model performance may vary depending on the image quality, scanner differences, and patient demographics. Models may inherit biases from the training data.
42
+ Training Details
43
+ Training Data
44
+ The models were trained on the Brain Tumor MRI Dataset from Kaggle. The dataset contains images categorized into three classes: glioma, meningioma, and pituitary tumors.
45
+ Training Hyperparameters
46
+
47
+ - Learning Rate: 0.0001
48
+ - Epochs: 25
49
+ - Batch Size: 32
50
+ - Optimizer: Adam
51
+ - Loss Function: Categorical Cross-Entropy
52
+
53
+ Evaluation
54
+ Testing Data
55
+ Test data is a stratified split of the original dataset with unseen examples from each tumor class.
56
+ Metrics
57
+ Accuracy, Precision, Recall, F1-score
58
+ Results
59
+
60
+ - ResNet50 Accuracy: ~98%
61
+ - MobileNetV2 Accuracy: ~96%
62
+
63
+ Environmental Impact
64
+
65
+ - Hardware Type: NVIDIA GPU (e.g., RTX 3060 or Colab T4)
66
+ - Hours Used: ~1-2 hours per model
67
+ - Cloud Provider: Google Colab
68
+ - Compute Region: [Not specified]
69
+ - Carbon Emitted: Estimated using ML Impact calculator (minimal due to short training duration)
70
+
71
+ Technical Specifications
72
+ Model Architecture and Objective
73
+ Transfer learning using pretrained CNNs (ResNet50 and MobileNetV2) adapted for multi-class classification of brain tumor MRI images.
74
+ Software
75
+ Python 3.10, TensorFlow/Keras, NumPy, Matplotlib, OpenCV