基于matplotlib的图像ROI的可视化

1
import roifile
2
from glob import glob
3
import matplotlib.pyplot as plt
4

5

6
def getImageFP(roi_fp):
7
    if roi_fp.endswith("_RoiSet.zip"):
8
        img_fp = roi_fp.replace("_RoiSet.zip", "")
9
    elif roi_fp.endswith("_rois.zip"):
10
        img_fp = roi_fp.replace("_rois.zip", ".png")
11
    return img_fp
12

13

14
def main(roi_fp):
15
    box = roifile.roiread(roi_fp)
16
    img_fp = getImageFP(roi_fp)
17
    img = io.imread(img_fp)
18
    ecs = {
19
        'add': '#ed1e79',
20
        'modify':'#f7931e',
21
        'delete':'#009245',
22
        'detect': '#0071bc'
23
    }
24
    fig, ax = plt.subplots()
25
    ax.imshow(img, cmap="Greys_r", vmin=0, vmax=255)
26
    for item in box:
27
        if item.name in ecs:
28
            ec = ecs[item.name]
29
        else:
30
            ec = ecs['detect']
31
        pts = item.coordinates()
32
        ax.fill(pts[:,0], pts[:,1], facecolor='none', edgecolor=ec, linewidth=1.5)
33
    ax.set_axis_off()
34
    plt.show()

1
wks = r'D:\desktop\data\'
2
rois = glob(wks+"/*/*.zip")
3
for roi_fp in rois:
4
    main(roi_fp)

1
def plotROI(img_fp, roi_fp, color='#0071bc', linewidth=3):
2
    box = roifile.roiread(roi_fp)
3
    img = io.imread(img_fp)
4
    fig, ax = plt.subplots()
5
    ax.imshow(img, cmap="Greys_r", vmin=0, vmax=255)
6
    for item in box:
7
        pts = item.coordinates()
8
        ax.fill(pts[:,0], pts[:,1],
9
                facecolor='none',
10
                edgecolor=color,
11
                linewidth=linewidth)
12
    ax.set_axis_off()
13
    plt.tight_layout()
14
    plt.show()