{ | |
"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] | |
} | |
} | |
} | |