brain-model-test / config.json
abdo1176's picture
Initial model upload with Keras, SavedModel, H5, weights, and configs
2a5e903 verified
raw
history blame contribute delete
280 Bytes
{
"model_type": "mobilevit",
"architecture": "MobileViT-XXS",
"num_classes": 2,
"input_shape": [
224,
224,
3
],
"class_names": [
"Healthy",
"Tumor"
],
"framework": "tensorflow",
"task": "image-classification",
"dataset": "brain-tumor-mri"
}