jherronr commited on
Commit
7150819
·
verified ·
1 Parent(s): d76a217

Subida del mejor modelo EfficientNet-B0 para clasificación de tumores cerebrales

Browse files
Files changed (3) hide show
  1. README.md +5 -0
  2. config.json +13 -0
  3. 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