Skip to content

Latest commit

 

History

History
26 lines (23 loc) · 591 Bytes

README.md

File metadata and controls

26 lines (23 loc) · 591 Bytes

Running the benchmark

  1. Install requirements
pip install -r requirements.txt
  1. Prepare a directory with images
  2. Run the benchmark
python benchmark.py --data-dir <path to directory with images> --images <number of images> --runs <number of runs> --print-package-versions

for example

python benchmark.py --data-dir '/hdd/ILSVRC2012_img_val' --images 2000 --runs 5 --print-package-versions

To use Pillow-SIMD instead of Pillow as a torchvision backend:

  1. Uninstall Pillow
pip uninstall -y pillow
  1. Install Pillow-SIMD
pip install pillow-simd