Siyeong-Lee / deep-hist

PyTorch implementation of Histogram Layers from DeepHist: Differentiable Joint and Color Histogram Layers for Image-to-Image Translation

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.