I’m trying to create a video by generating a sequence of 500 matplotlib plots, converting each to a numpy array, stacking them and then passing them to a SummaryWriter()'s add_video().  When I do this, the colorbar is converted from colored to black & white, and only a small number (~3-4) of the matplotlib plots are repeated. I confirmed that my numpy arrays are correct by using them to recreate a  matplotlib figure.
It’s difficult for me to create a small working example. Does anyone know what the problem might be?
My input tensor has shape (B,C,T,H,W), dtype np.uint8, and values between [0, 255].