随着分辨率的提高,整景遥感影像存储量也变大。由于计算机性能的限制(内存与GPU显存),在使用深度学习模型进行推理预测时,模型无法一次读入,需要分块将数据读入内存处理。在做目标识别推理时,分块可能会导致一个目标对象被切割到两个区块上,因此需要采用叠加滑动窗口的方法进行推理,但是这样又会导致同一个目标被识别两次,所以最后还需要对所有的识别结果进行NMS操作,去除重复识别的结果。
推理步骤
1、滑动叠加窗口分块读取
2、分块目标识别推理
3、所有识别结果合并后做NMS
4、将识别框的图像坐标转换为地理坐标,保存成shapefile文件。
代码
#!usr/bin/env python # -*- coding: utf-8 -*- """ @Author : zhaoguanhua @Email : @Time : 2021/7/13 23:09 @File : predict_all_img.py @Software: PyCharm """ import os,glob import numpy as np
讯享网

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容,请联系我们,一经查实,本站将立刻删除。
如需转载请保留出处:https://51itzy.com/kjqy/69190.html