在作者提供的代码(Github地址)的基础上,实现推理预测。
注意:首先得安装好CUDA、cuDNN以及GCC,下载地址
conda install pytorch torchvision -c pytorch
conda install matplotlib opencv pillow scikit-learn scikit-image
可能需要科学上网 源Github库下提供了的模型的下载地址,我将它们放在了本仓库的Releaes中,提供下载。
以dextr_pascal-sbd.pth
为例,下载后放到DEXTR/models
文件夹中。
net.py
主要提取源Github库中demo.py文件中的内容,并将其封装起来。
交互分割单张图像,通过OpenCV显示源图像,进行交互,显示分割结果,具体操作有:
- 运行程序,选择一张图像
- 左键进行交互(打点),右键撤销上次交互,当完成四个极值点的交互时,可以按下p键("P" or "p")进行分割,交互结果在显示的同时将保存在原图像的同意目录下。
- 按下o键("O" or "o")),选择一张新的照片进行分割。