disaster_images_model / preprocessor_config.json
Luwayy's picture
Create preprocessor_config.json
1615e64
raw
history blame contribute delete
787 Bytes
{
"do_normalize": true,
"do_resize": true,
"image_mean": [0.5, 0.5, 0.5],
"image_processor_type": "ViTImageProcessor",
"image_std": [0.5, 0.5, 0.5],
"size": {
"height": 224,
"width": 224
},
"image_formats": [".png", ".jpg", ".jpeg", ".bmp", ".gif", ".tiff"],
"do_verify_images": true,
"exclude_labels": ["Land_Slide"],
"resample_method": "bilinear",
"training_args": {
"batch_size": 16,
"num_epochs": 3,
"learning_rate": 3e-5,
"weight_decay": 0.01,
"evaluation_strategy": "epoch",
"save_strategy": "epoch"
},
"dataset_split": {
"train_test_ratio": 0.8
},
"image_augmentation": {
"resize": [224, 224],
"to_tensor": true,
"normalize": {
"mean": [0.5, 0.5, 0.5],
"std": [0.5, 0.5, 0.5]
}
}
}