现有的语义分割框架通常是针对普通的单张图片进行预测。如果要预测一张完整的大的遥感影像需要进行切片后预测最后拼接起来。这样预测比较费时费力费空间。通过阅读源码发现,paddleseg的预测功能是在paddleseg\core\predict.py中。因此可以修改这个文件实现一次性预测。本文改进了paddleseg的predict.py的代码。使用gdal分块读取方法,实现分块预测并写出为tif文件结构,效率高。
需要将本文件路径为paddleseg\core\predict.py。可以在代码 中修改预测分片的大小和重叠度。原有的predict.py可以重命名备份一下。