6d75aef13998df68bf940c01212b1826380832dc,main.py,,main,#,29

Before Change

    if TEST_BOUNDING:

        fig, ax = plt.subplots(1)
        x, y = denormalize(img.shape[1], 0, 0)
        ax.imshow(img)
        size = 64
        for i in range(3):

After Change

        fig, ax = plt.subplots(nrows=1, ncols=2)
        coords = denormalize(imgs.shape[1], loc)
        imgs = imgs.numpy()
        for i in range(len(imgs)):
            ax[i].imshow(imgs[i])
            size = 64
            for j in range(3):
                rect = bounding_box(coords[i, 0], coords[i, 1], size)
                ax[i].add_patch(rect)
                size = size * 2
            ax[i].get_xaxis().set_visible(False)
            ax[i].get_yaxis().set_visible(False)
        plt.savefig(plot_dir + "bbox.png", format="png", dpi=300,
                    bbox_inches="tight")
        plt.show()

Instances