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