pyrela/common_utils/helper.py [146:164]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ax[r, c].axis("off")

        if data.shape[0] == 3:
            data = data.swapaxes(0, 1).swapaxes(1, 2)
            ax[r, c].imshow(data, vmin=0, vmax=1)
            continue

        # print('>>>', data.shape)
        # if data.shape[0] > 3:
        #     data = data[0]

        # print(data.shape)
        ax[r, c].imshow(data, vmin=0, vmax=1, cmap="gray")
        # ax[r, c].set_title('c%d_%s' % (i, channel_names[i]), fontsize=50)
        # break

    plt.tight_layout()
    plt.savefig(path)
    plt.close()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



pyrela/common_utils/helper.py [183:201]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ax[r, c].axis("off")

        if data.shape[0] == 3:
            data = data.swapaxes(0, 1).swapaxes(1, 2)
            ax[r, c].imshow(data, vmin=0, vmax=1)
            continue

        # print('>>>', data.shape)
        # if data.shape[0] > 3:
        #     data = data[0]

        # print(data.shape)
        ax[r, c].imshow(data, vmin=0, vmax=1, cmap="gray")
        # ax[r, c].set_title('c%d_%s' % (i, channel_names[i]), fontsize=50)
        # break

    plt.tight_layout()
    plt.savefig(path)
    plt.close()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -