Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ language:
|
|
30 |
pipeline_tag: image-classification
|
31 |
---
|
32 |
|
33 |
-
|
34 |
|
35 |
This model is a fine-tuned version of [facebook/deit-base-distilled-patch16-224](https://huggingface.co/facebook/deit-base-distilled-patch16-224).
|
36 |
|
@@ -47,41 +47,41 @@ It achieves the following results on the evaluation set:
|
|
47 |
- Micro precision: 0.8046
|
48 |
- Macro precision: 0.8710
|
49 |
|
50 |
-
|
51 |
-
|
|
|
|
|
52 |
<a href="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/tree/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison">
|
53 |
Click here for the code that I used to create this model
|
54 |
</a>
|
55 |
-
|
56 |
-
|
57 |
-
Click here to see the README markdown file for the full project
|
58 |
</a>
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
This model is intended to demonstrate my ability to solve a complex problem using technology.
|
63 |
|
64 |
-
## Training and evaluation data
|
65 |
-
|
66 |
-
<div style="align: center">
|
67 |
-
<a href="https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri" span="">
|
68 |
-
Brain Tumor Image Classification Dataset
|
69 |
-
</a>
|
70 |
-
|
71 |
-
Sample Images
|
72 |
-
|
73 |
-
![Sample Images](https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Sample%20Images.png)
|
74 |
-
|
75 |
-
<strong>Class Distribution</strong>
|
76 |
-
|
77 |
-
<em>Training Dataset</em>
|
78 |
-
|
79 |
-
![Training Dataset](https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Class%20Distribution%20-%20Training%20Dataset.png)
|
80 |
-
|
81 |
-
<em>Evaluation Dataset</em>
|
82 |
-
|
83 |
-
![Evaluation Dataset](https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Class%20Distribution%20-%20Testing%20Dataset.png)
|
84 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
85 |
</div>
|
86 |
|
87 |
## Training procedure
|
|
|
30 |
pipeline_tag: image-classification
|
31 |
---
|
32 |
|
33 |
+
<h1>deit-base-distilled-patch16-224-Brain_Tumors_Image_Classification</h1>
|
34 |
|
35 |
This model is a fine-tuned version of [facebook/deit-base-distilled-patch16-224](https://huggingface.co/facebook/deit-base-distilled-patch16-224).
|
36 |
|
|
|
47 |
- Micro precision: 0.8046
|
48 |
- Macro precision: 0.8710
|
49 |
|
50 |
+
<div style="text-align: center;">
|
51 |
+
<h2>
|
52 |
+
Model Description
|
53 |
+
</h2>
|
54 |
<a href="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/tree/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison">
|
55 |
Click here for the code that I used to create this model
|
56 |
</a>
|
57 |
+
This project is part of a comparison of seventeen (17) transformers.
|
58 |
+
<a href="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/README.md">
|
59 |
+
Click here to see the README markdown file for the full project
|
60 |
</a>
|
61 |
+
<h2>
|
62 |
+
Intended Uses & Limitations
|
63 |
+
</h2>
|
64 |
This model is intended to demonstrate my ability to solve a complex problem using technology.
|
65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
|
67 |
+
<h2>
|
68 |
+
Training & Evaluation Data
|
69 |
+
</h2>
|
70 |
+
<a href="https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri">
|
71 |
+
Brain Tumor Image Classification Dataset
|
72 |
+
</a>
|
73 |
+
<h2>
|
74 |
+
Sample Images
|
75 |
+
</h2>
|
76 |
+
<img src="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Sample%20Images.png" />
|
77 |
+
<h2>
|
78 |
+
Class Distribution of Training Dataset
|
79 |
+
</h2>
|
80 |
+
<img src="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Class%20Distribution%20-%20Training%20Dataset.png"/>
|
81 |
+
<h2>
|
82 |
+
Class Distribution of Evaluation Dataset
|
83 |
+
</h2>
|
84 |
+
<img src="https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/raw/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Brain%20Tumors%20Image%20Classification%20Comparison/Images/Class%20Distribution%20-%20Testing%20Dataset.png"/>
|
85 |
</div>
|
86 |
|
87 |
## Training procedure
|