sensei-ml
/

brain_tumors_detection

Model card Files Files and versions

sensei-ml commited on Jun 4, 2025

Commit

139a495

·

verified ·

1 Parent(s): e4da73b

Update config.json

Files changed (1) hide show

config.json +4 -3

config.json CHANGED Viewed

@@ -1,9 +1,9 @@
 {
-  "model_type": "resnet",
   "architectures": [
-    "ResNetForObjectDetection"
   ],
-  "framework": "pytorch",
   "image_size": 224,
   "num_labels": 2,
   "id2label": {
@@ -17,3 +17,4 @@
   "task": "object-detection",
   "object_detection": true
 }

 {
+  "model_type": "detr",
   "architectures": [
+    "DetrForObjectDetection"
   ],
+  "backbone": "resnet50",
   "image_size": 224,
   "num_labels": 2,
   "id2label": {
   "task": "object-detection",
   "object_detection": true
 }