Subida del mejor modelo EfficientNet-B0 para clasificación de tumores cerebrales
Browse files- README.md +5 -0
- config.json +13 -0
- pytorch_model.bin +3 -0
README.md
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# EfficientNet-B0 Brain Tumor Classifier
|
| 2 |
+
|
| 3 |
+
Modelo fine-tuned para clasificar imágenes MRI en tres tipos de tumor cerebral: brain_glioma, brain_menin, brain_tumor.
|
| 4 |
+
|
| 5 |
+
Entrenado con PyTorch y transfer learning.
|
config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model": "efficientnet_b0",
|
| 3 |
+
"num_labels": 3,
|
| 4 |
+
"classes": [
|
| 5 |
+
"brain_glioma",
|
| 6 |
+
"brain_menin",
|
| 7 |
+
"brain_tumor"
|
| 8 |
+
],
|
| 9 |
+
"image_size": [
|
| 10 |
+
224,
|
| 11 |
+
224
|
| 12 |
+
]
|
| 13 |
+
}
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f781137dc1bf2a7f65034e8a127c5cbdcfe6141a5119b2364efa006ae8584c0
|
| 3 |
+
size 16345770
|