REVA-QCAV/data_vis.py

import matplotlib.pyplot as plt


def plot_img_mask(img, mask):
    fig = plt.figure()

    ax1 = fig.add_subplot(1, 3, 1)
    ax1.imshow(img)

    ax2 = fig.add_subplot(1, 3, 2)
    ax2.imshow(mask)

    plt.show()
First prototype slow, no gpu, no validation 2017-08-16 12:24:29 +00:00			`import matplotlib.pyplot as plt`

Created a basic train loop + changed a bit loss and utils 2017-08-17 19:16:19 +00:00
First prototype slow, no gpu, no validation 2017-08-16 12:24:29 +00:00			`def plot_img_mask(img, mask):`
			`fig = plt.figure()`

			`ax1 = fig.add_subplot(1, 3, 1)`
			`ax1.imshow(img)`

			`ax2 = fig.add_subplot(1, 3, 2)`
			`ax2.imshow(mask)`

			`plt.show()`