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