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.gitcd /d D:cd EISeg-developpip 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