22_提取建筑-iObjects Python with JupyterHub for K8s

原创
2019/11/11 11:44
阅读数 111

提取建筑-iObjects Python with JupyterHub for K8s

import os
import time

from iobjectspy import open_datasource
from iobjectspy.ai.recognition import extract_region
---------------------------------------------------------------------------

ModuleNotFoundError                       Traceback (most recent call last)

<ipython-input-1-a1944dbfe13b> in <module>()
      3 
      4 from iobjectspy import open_datasource
----> 5 from iobjectspy.ai.recognition import extract_region


/opt/conda/lib/python3.6/site-packages/iobjectspy/ai/recognition.pyc in <module>()


ModuleNotFoundError: No module named 'rasterio'
data_dir = ''
out_dir = os.path.join(data_dir, 'out/')
model_path = os.path.join(curr_dir, 'model')

if not os.path.exists(out_dir):
    os.makedirs(out_dir)

    
def extract_building_file_dom():
    """
    基于DOM影像文件进行建筑物提取,结果面数据集保存在UDB数据源下
    """

    print('使用CPU进行提取约需10分钟,使用GPU提取约需1分钟')

    dom_path = os.path.join(data_dir, 'dom.tif')
    start_time = time.time()
    extract_region(dom_path, model_path, model_name='building_extract_with_dom',
                   output=os.path.join(out_dir, 'out_building.udb'), out_dataset_name='file_dom')
    end_time = time.time()
    print('提取完成,共耗时{}s,结果数据保存在 {} 数据源下 file_dom 数据集中'.format(
        end_time - start_time, out_dir + 'out_building.udb'))
if __name__ == '__main__':

    # 基于DOM影像文件进行建筑物提取
    extract_building_file_dom()
exit()
展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部