这里记录一下跑通的一些交互式图像分割、图像实例分割的代码。
在一些有给出相关代码（有相应的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，下载地址