Upload 236 files

.gitattributes

@@ -25,7 +25,6 @@
 *.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
 *.wasm filter=lfs diff=lfs merge=lfs -text
@@ -33,3 +32,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
+variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,105 @@
+---
+license: apache-2.0
+language:
+- de
+- en
+metrics:
+- accuracy
+- roc_auc
+library_name: keras
+pipeline_tag: image-classification
+---
+# Brain tumor classification using CNN
+
+<!-- Provide a quick summary of what the model is/does. 
+
+This modelcard aims to be a base template for new models. It has been generated using [this raw template](https://github.com/huggingface/huggingface_hub/blob/main/src/huggingface_hub/templates/modelcard_template.md?plain=1).
+-->
+## Model Details
+
+### Model Description
+
+```python
+model.summary()
+Out:
+Model: "sequential"
+_________________________________________________________________
+ Layer (type)                Output Shape              Param #   
+=================================================================
+ rescaling (Rescaling)       (None, 200, 200, 1)       0         
+                                                                 
+ conv2d (Conv2D)             (None, 200, 200, 16)      160       
+                                                                 
+ max_pooling2d (MaxPooling2D  (None, 100, 100, 16)     0         
+ )                                                               
+                                                                 
+ conv2d_1 (Conv2D)           (None, 100, 100, 32)      4640      
+                                                                 
+ max_pooling2d_1 (MaxPooling  (None, 50, 50, 32)       0         
+ 2D)                                                             
+                                                                 
+ conv2d_2 (Conv2D)           (None, 50, 50, 64)        18496     
+                                                                 
+ max_pooling2d_2 (MaxPooling  (None, 25, 25, 64)       0         
+ 2D)                                                             
+                                                                 
+ flatten (Flatten)           (None, 40000)             0         
+                                                                 
+ dense (Dense)               (None, 128)               5120128   
+                                                                 
+ dense_1 (Dense)             (None, 64)                8256      
+                                                                 
+ dense_2 (Dense)             (None, 128)               8320      
+                                                                 
+ dense_3 (Dense)             (None, 64)                8256      
+                                                                 
+ dense_4 (Dense)             (None, 32)                2080      
+                                                                 
+ dense_5 (Dense)             (None, 96)                3168      
+                                                                 
+ dense_6 (Dense)             (None, 96)                9312      
+                                                                 
+ dense_7 (Dense)             (None, 128)               12416     
+                                                                 
+ dense_8 (Dense)             (None, 1)                 129       
+                                                                 
+=================================================================
+Total params: 5,195,361
+Trainable params: 5,195,361
+Non-trainable params: 0
+_________________________________________________________________
+
+```
+
+### Dataset
+The dataset is composed of [Brain Tumor Classification (MRI)](https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri) and
+[Brain MRI Images for Brain Tumor Detection](https://www.kaggle.com/datasets/navoneel/brain-mri-images-for-brain-tumor-detection)
+
+Using image data augmentation we get 199.632 files belonging to 2 classes.
+
+train-test-split: 80/20 
+
+Training (159705 files):
+  - Using 143735 files for training
+  - Using 15970 files for validation
+
+Test/Validation:
+  - 39927 files
+
+### Training
+Coming Soon
+
+### Validation
+Coming Soon
+
+#### Hardware
+
+Lenovo Thinkpad P14s
+  - CPU (# Cores/Threads): AMD Ryzen 7 PRO 5850U (8/16)
+  - RAM: 32 GB
+
+#### Software
+
+DataSpell 2023.1.2
+  - Python 3.10.9
+  - Tensorflow 2.12.0

config.json

@@ -0,0 +1,14 @@
+{
+  "_name_or_path": "cnn_brain_tumor_classification",
+  "architectures": [
+    "CNNForBinaryClassification"
+  ],
+  "id2label": {
+    "0": "No",
+    "1": "Yes"
+  },
+  "label2id": {
+    "No": 0,
+    "Yes": 1
+  }
+}

keras_metadata.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44e4a5a364697dc747db1da2f6b4f67c6be46ef11b5616ebb17cd715c38ea1a5
+size 35537

saved_model.pb

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