Visualize in breakpoint 1

[fengziyue]

Hi @SeokjuLee :

Thank you for sharing this amazing work, the visualization in this code helps me a lot in understanding the logic. However, I have a confusion, the line below:

Insta-DM/train.py

Line 402 in 95fac7f

fig.add_subplot(ea1,ea2,ii); ii += 1; plt.imshow(r2t_inst); plt.colorbar(); plt.text(10, -14, "r2t_inst", fontsize=7, bbox=dict(facecolor='None', edgecolor='None'));

will trigger the error
"TypeError: Invalid shape (256, 832, 2) for image data"

There are two instance masks in r2t_inst, are you supposed to visualize one of them like r2t_inst[:, :, 0] or do you have other implementations to visualize both of them in once?

Thank you!