Fields of The World (FTW)|机器学习数据集|农业数据集
收藏Fields of The World (FTW) - Baselines Codebase
概述
Fields of The World (FTW) 是一个大规模基准数据集,旨在推动农业田地边界实例分割的机器学习模型的发展。该数据集支持全球农业监测、土地利用评估和环境研究中对准确和可扩展的田地边界数据的需要。
目录结构
Fields-of-The-World ├── .flake8 ├── .gitignore ├── CHANGELOGS.md ├── LICENSE ├── README.md ├── assets ├── configs │ └── example_config.yaml ├── environment.yml ├── inference.py ├── notebooks │ └── visualize_dataset.ipynb ├── pyproject.toml └── src ├── ftw │ ├── init.py │ ├── datamodules.py │ ├── datasets.py │ ├── metrics.py │ ├── trainers.py │ └── utils.py └── ftw_cli ├── init.py ├── cli.py ├── download.py ├── model.py └── unpack.py
系统设置
创建 Conda/Mamba 环境
bash mamba env create -f env.yml mamba activate ftw
验证 PyTorch 安装和 CUDA 可用性
bash python -c "import torch; print(torch.cuda.is_available())"
设置 FTW CLI
bash pip install -e .
数据集设置
下载并解压压缩版本
bash ftw download --help ftw unpack --help
数据集可视化
使用 visualize_dataset.ipynb 探索数据集。
实验准备
在运行实验之前,确保在 configs 目录中创建配置文件,并更新 datasets.py 中的 root 参数以反映正确的数据集路径。
实验
训练
从头开始训练模型
bash ftw model fit --config configs/example_config.yaml
从检查点恢复训练
bash ftw model fit --config configs/example_config.yaml --ckpt_path <Checkpoint File Path>
测试
bash ftw model test --gpu 0 --checkpoint_fn logs/path_to_model/checkpoints/last.ckpt --countries denmark finland --postprocess --output_fn results.csv
并行实验
使用 run_experiments.py 脚本在多个 GPU 上并行运行实验。
贡献
欢迎贡献!请 fork 仓库,进行更改并提交拉取请求。
许可证
该代码库在 MIT 许可证下发布。详见 LICENSE 文件。




