EISeg(Efficient Interactive Segmentation)是以RITM及EdgeFlow算法为基础,基于飞桨开发的一个高效智能的交互式分割标注软件。涵盖了通用、人像、遥感、医疗、工业质检等不同方向的高质量交互式分割模型,方便开发者快速实现语义及实例标签的标注,降低标注成本。 另外,将EISeg获取到的标注应用到PaddleSeg提供的其他分割模型进行训练,便可得到定制化场景的高精度模型,打通分割任务从数据标注到模型训练及预测的全流程。

安装步骤

1. 创建并激活虚拟环境(建议新建虚拟环境)

conda create -n eiseg python=3.8
conda activate eiseg

2. 安装paddlepaddle环境

这里默认安装的CPU版的,如果要安装GPU版,看 paddlepaddle 官方文档:https://www.paddlepaddle.org.cn/

conda install paddlepaddle==2.5.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

3. 安装eiseg(两种方式)

  • pip安装

    pip install -i https://pypi.tuna.tsinghua.edu.cn/simple eiseg --trusted-host pypi.tuna.tsinghua.edu.cn
    

    若启动执行python -m eiseg报以下错误:

    AttributeError: module 'cv2.dnn' has no attribute 'DictValue'
    

    卸载opencv-python,安装opencv-contrib-python

    pip uninstall opencv-python
    pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-contrib-python --trusted-host pypi.tuna.tsinghua.edu.cn
    

    在虚拟环境中输入python -m eiseg即可运行。

  • 源码
    参照EISeg开源网址:https://github.com/PaddlePaddle/PaddleSeg/tree/release/2.5/EISeg

     git clone https://github.com/PaddleCV-SIG/EISeg.git
     cd /d D:
     cd EISeg-develop
     pip install -r requirements.txt
    

运行
python -m eiseg

4. 模型准备

  • 需加载*.pdiparams模型文件

    模型下载路径:https://github.com/PaddlePaddle/PaddleSeg/tree/release/2.5/EISeg#%E6%A8%A1%E5%9E%8B%E5%87%86%E5%A4%87

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐