REVA-QCAV/utils/data_vis.py

import matplotlib.pyplot as plt

def plot_img_and_mask(img, mask):
    fig = plt.figure()
    a = fig.add_subplot(1, 2, 1)
    a.set_title('Input image')
    plt.imshow(img)

    b = fig.add_subplot(1, 2, 2)
    b.set_title('Output mask')
    plt.imshow(mask)
    plt.show()
First prototype slow, no gpu, no validation 2017-08-16 12:24:29 +00:00			`import matplotlib.pyplot as plt`

Migration to PyTorch 0.4, code cleanup Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c 2018-06-08 17:27:32 +00:00			`def plot_img_and_mask(img, mask):`
First prototype slow, no gpu, no validation 2017-08-16 12:24:29 +00:00			`fig = plt.figure()`
Migration to PyTorch 0.4, code cleanup Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c 2018-06-08 17:27:32 +00:00			`a = fig.add_subplot(1, 2, 1)`
			`a.set_title('Input image')`
			`plt.imshow(img)`

			`b = fig.add_subplot(1, 2, 2)`
			`b.set_title('Output mask')`
			`plt.imshow(mask)`
			`plt.show()`