Python中用于循环的子图

时间:2018-11-13 19:31:15

标签: python matplotlib

我从SQL搜索中收到了一个字典,然后我想遍历字典以在x,y,h,w上裁剪图像。然后,我想显示原始图像,即编辑后的图像。我已经完成了。我无法弄清楚的是如何使用matplotlib.pyplot在图中显示结果。这是我的代码:

sql_result = get_rois(config, "145", "a")

img_dir = '../imgs/'

make palaeographical chart
fig = plt.figure()

#Iterate over the list
for num, result in enumerate(aleph_rois):
    #make path to image
    img_path = os.path.join(imgs, sql_result['file_name'])

    #declare variables for cropping character
    x = sql_result['x']
    y = sql_result['y']
    w = sql_result['w']
    h = sql_result['h']

    #crop to character
    img = cv2.imread(img_path, 0)
    crop = img[y:y+h, x:x+w]
    th0 = cv2.adaptiveThreshold(crop, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 21, 2)

    #OTSU Binarization to fill
    blur = cv2.GaussianBlur(crop,(5,5),0)
    ret, th1 = cv2.threshold(blur,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)

    frg_loc = str(sql_result['artefact_siglum']) + " " + str(result['line_id']) + " roi:" + str(sql_result['roi_id'])

    #create fig
    plt.title(frg_loc)
    plt.imshow(th1, cmap='Greys_r')

以上显然是失败的。我尝试了几种选择,但我认为图像存在图像相互覆盖的问题。但是目标是使th0和th1处于2列图中,并且行数由sql_results中的结果数定义。很感谢任何形式的帮助。谢谢。

1 个答案:

答案 0 :(得分:0)

我不确定我是否理解正确-但是,至少作为讨论的开始:

import matplotlib.pyplot as plt

N = 3
fig, axs = plt.subplots(N, 2)
for row in range(N):
    axs[row, 0].imshow(th0)
    axs[row, 1].imshow(th1, cmap = 'Reds_r')

产生

enter image description here

(图像内容仅是示例性的,而不是此SO Answer TM 的代码的一部分)