https://github.com/lusinlu/image-entropy-gpu
Calculation of the entropy of the batch of images (whole image or patches)
https://github.com/lusinlu/image-entropy-gpu
computer-vision deep-learning entropy entropy-measures gpu pytorch
Last synced: 9 months ago
JSON representation
Calculation of the entropy of the batch of images (whole image or patches)
- Host: GitHub
- URL: https://github.com/lusinlu/image-entropy-gpu
- Owner: lusinlu
- Created: 2021-10-15T12:08:41.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2021-10-15T13:21:29.000Z (over 4 years ago)
- Last Synced: 2025-04-04T10:30:46.933Z (about 1 year ago)
- Topics: computer-vision, deep-learning, entropy, entropy-measures, gpu, pytorch
- Language: Python
- Homepage:
- Size: 4.88 KB
- Stars: 9
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# image_entropy_GPU
Script for calculating the entropy of the batch of images on GPU with Pytorch.
Two options are available
1. Calculate entropy of the whole image - pass `--patch_size 0`
2. Calculate entropy of patches in the image - pass `--patch_size 'size of the patch'` (default is 64)
## Requirements
for installing required packages run
` pip install -r requirements.txt`
## Usage
To test the code run
`python main.py --data_path 'path to images' `
## Acknowledgement
Calculation of the pdf function is modified from [kornia](https://github.com/kornia/kornia).