File size: 817 Bytes
f5997ce f8a9cee f5997ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | {
"model_type": "calorie-clip",
"version": "1.0.0",
"base_model": "ViT-B-32",
"pretrained": "openai",
"hidden_dim": 512,
"output_dim": 1,
"task": "calorie_regression",
"training": {
"epochs": 30,
"batch_size": 16,
"learning_rate_clip": 1e-5,
"learning_rate_head": 1e-3,
"loss_function": "huber",
"optimizer": "adamw",
"fine_tuned_layers": "last_2_transformer_blocks"
},
"performance": {
"mae": 51.4,
"within_50_cal": 67.6,
"within_100_cal": 90.5,
"test_samples": 1951,
"training_samples": 11053
},
"preprocessing": {
"image_size": 224,
"mean": [0.48145466, 0.4578275, 0.40821073],
"std": [0.26862954, 0.26130258, 0.27577711]
},
"license": "MIT",
"author": "Haplo LLC",
"intended_use": "Food calorie estimation from images"
}
|