这里记录一下跑通的一些交互式图像分割、图像实例分割的代码。
在一些有给出相关代码(有相应的Github库)的基础上,封装一个推理预测的类(net.py),然后在Pascal VOC 2012数据集(官方地址 host.robots.ox.ac.uk、Pascal VOC Dataset Mirror (pjreddie.com)包含测试集)上,测试(test.py)得到分割结果(mask),并计算相关指标(IOU、F1-Score)。
- net.py:预测推理,包含图像预处理(normalize等)、后处理(二值化等)。
- test.py:在Pascal VOC 2012数据集上,做测试得到分割结果(mask),并计算相关指标(IOU、F1-Score)。
封装一个推理预测的类(net.py
),并通过用户交互进行分割测试(test.py
)。
二、交互式图像分割
A survey of recent interactive image segmentation methods, PDF - springer.com
网络 | 来源/源Github库 | 本库中地址 | 相关描述 | |
---|---|---|---|---|
DeepGrabCut | jfzhang95/DeepGrabCut-PyTorch | DeepGrabCut | - | 打包下载 |
DEXTR | scaelles/DEXTR-PyTorch | DEXTR | - | 打包下载 |
G99AIS | MarcoForte/DeepInteractiveSegmentation | G99AIS | - | 打包下载 |
IIS-LD | intel-isl/Intseg | IIS-LD | Tensorflow 1.x | 打包下载 |
FCTSFN | cyh4/FCTSFN | FCTSFN | Caffe (通过OpenCV dnn 实现推理预测) |
打包下载 |
BRS | wdjang/BRS-Interactive_segmentation | BRS | Caffe (通过OpenCV dnn 实现推理预测) |
打包下载 |
FBRS | saic-vul/fbrs_interactive_segmentation | FBRS | - | 打包下载 |
FCANet | frazerlin/fcanet | FCANet | - | 打包下载 |
一些根据图像标签随机采点的随机采样方法
三、图像实例分割
A Survey on Instance Segmentation: State of the art, PDF - arcix.org
网络 | 来源/源Github库 | 修改后的代码 | 相关描述 | |
---|---|---|---|---|
RSIS | imatge-upc/rsis | RSIS | - | 打包下载 |
EmbedMask | yinghdb/EmbedMask | EmbedMask | - | 打包下载 |
RPEIG | aimerykong/ Recurrent-Pixel-Embedding-for-Instance-Grouping |
RPEIG | Matlab代码 | 打包下载 |
MaskRCNN | torchvision | MaskRCNN | - | 打包下载 |
RefineMask | zhanggang001/RefineMask | RefineMask | - | 打包下载 |
CenterMask | youngwanLEE/CenterMask | CenterMask | - | 打包下载 |
D2Det | JialeCao001/D2Det | D2Det | - | 打包下载 |
windows 10 (20H2)
VS Community 2017 (15.9.36)
Matlab 2018b
CUDA 10.2.89 / 11.2.142,下载地址
cuDNN 7.6.5 / 8.0.5,下载地址
MinGW-W64 GCC-5.4.0,下载地址