I visualize the img2 in max_ssim.py, but it is not the gif shown in the home page. what mistake do I make? The following is the code, modified in max_ssim.py.

[THHHomas]

while ssim_value < 0.95:
optimizer.zero_grad()
ssim_out = -ssim_loss(img1, img2)
ssim_value = - ssim_out.item()
print(ssim_value)
ssim_out.backward()
optimizer.step()
ii = np.transpose(img2[0].detach().cpu().numpy(), (1,2,0))
cv2.imshow("pre", ii255)
print(np.max(ii255))
cv2.waitKey(100)

[bornabesic]

You haven't explained what kind of results you get, but I guess you are looking at a noisy white image. After you multiply the array by 255 you also need to convert it to a byte type.

cv2.imshow("pre", (ii * 255).astype(np.uint8))