Deep Image Structure and Texture Similarity (DISTS) Metric
from DISTS_pytorch import DISTS dists = DISTS() # calculate DISTS between X, Y (a batch of RGB images, data range: 0~1) # X: (N,3,H,W) # Y: (N,3,H,W) dists_value = dists(X, Y) # set 'require_grad=True, batch_average=True' to get a scalar value as loss. dists_loss = dists(X, Y, require_grad=True, batch_average=True) dists_loss.backward()
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size DISTS_pytorch-1.0.0.tar.gz (11.1 kB)||File type Source||Python version None||Upload date||Hashes View hashes|