| { |
| "accelerator": "auto", |
| "annotations_xml": null, |
| "architecture": "retinanet", |
| "batch_size": 1, |
| "cropmodel": { |
| "balance_classes": false, |
| "batch_size": 4, |
| "label_dict": { |
| "Alive": 0, |
| "Dead": 1 |
| }, |
| "lr": 0.0001, |
| "num_classes": 2, |
| "num_workers": 0, |
| "sampler": "weighted_random", |
| "scheduler": { |
| "params": { |
| "T_max": 10, |
| "cooldown": 0, |
| "eps": 1e-08, |
| "eta_min": 1e-05, |
| "factor": 0.5, |
| "gamma": 0.1, |
| "lr_lambda": "0.95 ** epoch", |
| "milestones": [ |
| 50, |
| 100 |
| ], |
| "min_lr": 0.0, |
| "mode": "min", |
| "patience": 5, |
| "step_size": 30, |
| "threshold": 0.0001, |
| "threshold_mode": "rel" |
| }, |
| "type": "ReduceLROnPlateau" |
| } |
| }, |
| "devices": "auto", |
| "label_dict": { |
| "Tree": 0 |
| }, |
| "model": { |
| "name": "weecology/deepforest-tree", |
| "revision": "main" |
| }, |
| "nms_thresh": 0.05, |
| "num_classes": 1, |
| "patch_overlap": 0.05, |
| "patch_size": 400, |
| "path_to_raster": null, |
| "path_to_rgb": null, |
| "predict": { |
| "pin_memory": false |
| }, |
| "rgb_dir": null, |
| "score_thresh": 0.1, |
| "train": { |
| "augmentations": [ |
| { |
| "HorizontalFlip": { |
| "p": 0.5 |
| } |
| } |
| ], |
| "csv_file": null, |
| "epochs": 1, |
| "fast_dev_run": false, |
| "lr": 0.001, |
| "preload_images": false, |
| "root_dir": null, |
| "scheduler": { |
| "params": { |
| "T_max": 10, |
| "cooldown": 0, |
| "eps": 1e-08, |
| "eta_min": 1e-05, |
| "factor": 0.1, |
| "gamma": 0.1, |
| "lr_lambda": "0.95 ** epoch", |
| "milestones": [ |
| 50, |
| 100 |
| ], |
| "min_lr": 0.0, |
| "mode": "min", |
| "patience": 10, |
| "step_size": 30, |
| "threshold": 0.0001, |
| "threshold_mode": "rel" |
| }, |
| "type": null |
| } |
| }, |
| "validation": { |
| "augmentations": [], |
| "csv_file": null, |
| "iou_threshold": 0.4, |
| "lr_plateau_target": "val_loss", |
| "preload_images": false, |
| "root_dir": null, |
| "size": null, |
| "val_accuracy_interval": 20 |
| }, |
| "workers": 0 |
| } |
