layout-generator 生成数据集配套 Dataset 封装，包含 pytorch-lightning 封装的 FPN 模型，可以方便的配置参数。
- LoadResponse
- LoadResponseH5
- LayoutDataset
- LayoutDatasetH5
git clone https://git.idrl.site/zweien/layout-data.git
cd layout-data
pip install -U -e .example目录下包含基本示例train.py: 训练模型主程序config.yml: 参数配置文件
- 使用
python train.py -h查看支持的参数列表data_dir为数据集目录，该目录下应包含train与test子目录， 使用 layout-generator 生成相应数据
python train.py --config config.yml开始训练- 默认使用
tensorboard记录中间训练数据
结果请参考 example