20241204124
立即下载
资源介绍:
20241204124
# deep-hist
PyTorch implementation of Histogram Layers from DeepHist: Differentiable Joint and Color Histogram Layers for Image-to-Image Translation
PyTorch modules are ready to be used (metrics & histogram layers).
Run hist_loss to produce plots and metrics for example images.
The official implementation using TensorFlow can be found [here](https://github.com/mor-avi-aharon-bgu/HueNet).
Please cite the original publication if you use this code in your work:
```
M. Avi-Aharon, A. Arbelle and T. R. Raviv,
"Differentiable Histogram Loss Functions for Intensity-based Image-to-Image Translation,"
in IEEE Transactions on Pattern Analysis and Machine Intelligence,
vol. 45, no. 10, pp. 11642-11653, Oct. 2023, doi: 10.1109/TPAMI.2023.3278287.
```