Go to file
Daniel Løvbrøtte Olsen 7db05f6565 Update rank_photos.py 2019-10-31 17:49:38 +00:00
.gitignore Create .gitignore 2019-04-30 11:44:02 -05:00
README.md Fixes for Py3 2019-04-25 09:37:03 -05:00
README.rst converting .png to .jpg 2016-02-20 12:42:49 -08:00
rank_photos.py Update rank_photos.py 2019-10-31 17:49:38 +00:00
screenshot.jpg converting .png to .jpg 2016-02-20 12:42:49 -08:00

README.md

Photo Ranking With Python

Screenshot

What is this?

This is a tool that uses the Elo Ranking System written in Python using:

  1. Matplotlib
  2. Numpy
  3. Pillow
  4. exifread

Features:

  • Auto image rotation that the camera recored in the EXIF meta data
  • Persistent state from execution to execution so you can pickup where you left off
  • New photos that are added to the photo dir after initial ranking are picked up

Install dependencies

Use your system's package manager to install Matplotlib & Numpy if you don't already have them installed.

$ pip install Matplotlib [--user] <---------------------This should also install Numpy

Next, you can use pip to install the EXIF image reader package exifread.

$ pip install exifread [--user] - The --user flag is optional and it depends on your environment.

Note for Windows Users - You may need to install Pillow for jpg support. $ pip install Pillow [--user]

How to rank photos

Once you have to dependencies installed, run rank_photos.py on the command line passing it the directory of photos.

    $ ./rank_photos.py -h

    usage: rank_photos.py [-h] [-r N_ROUNDS] [-f FIGSIZE FIGSIZE] photo_dir

    Uses the Elo ranking algorithm to sort your images by rank. The program globs
    for .jpg images to present to you in random order, then you select the better
    photo. After n-rounds, the results are reported.

    positional arguments:
      photo_dir             The photo directory to scan for .jpg images

    optional arguments:
      -h, --help            show this help message and exit
      -r N_ROUNDS, --n-rounds N_ROUNDS
                            Specifies the number of rounds to pass through the
                            photo set (3)
      -f FIGSIZE FIGSIZE, --figsize FIGSIZE FIGSIZE
                            Specifies width and height of the Matplotlib figsize
                            (20, 12)

For example, iterate over all photos three times:

$ ./rank_photos.py -r 3 ~/Desktop/example/

After the number of rounds complete, ranked.txt is written into the photo dir.

Ranking work is cached

After completing N rounds of ranking, a file called ranking_table.json is written into the photo dir. The next time rank_photos.py is executed with the photo dir, this table is read in and ranking can continue where you left off.

You can also add new photos the the directory and they will get added to the ranked list even though they weren't included previously.

Example

Suppose there is a dir containing some photos:

$ ls -1 ~/Desktop/example/

    20160102_164732.jpg
    20160109_151557.jpg
    20160109_151607.jpg
    20160109_152318.jpg
    20160109_152400.jpg
    20160109_152414.jpg
    20160109_153443.jpg

The photos haven't been ranked yet. Let's rank them in 1 round, shall we?

$ ./rank_photos.py -r 1 ~/Desktop/example/

Once the number of rounds is complete, the ranked list is dumped to the console:

    Final Ranking:
    Rank    Score    Matches    Win %    Filename
       1    1433          2     100.00    20160109_152414.jpg
       2    1414          3      66.67    20160109_151557.jpg
       3    1401          2      50.00    20160109_153443.jpg
       4    1400          2      50.00    20160102_164732.jpg
       5    1387          3      33.33    20160109_151607.jpg
       6    1383          3      33.33    20160109_152318.jpg
       7    1382          3      33.33    20160109_152400.jpg

The ranked list is also written to the file ranked.txt. The raw data is cached to the file ranking_table.json:


    $ cat ~/Desktop/example/ranking_table.json

    {
        "photos" : [
            {
                "matches" : 2,
                "wins" : 2,
                "score" : 1432.736306793522,
                "filename" : "20160109_152414.jpg"
            },
            {
                "matches" : 3,
                "wins" : 2,
                "score" : 1413.760501639972,
                "filename" : "20160109_151557.jpg"
            },
            {
                "matches" : 2,
                "wins" : 1,
                "score" : 1400.736306793522,
                "filename" : "20160109_153443.jpg"
            },
            {
                "matches" : 2,
                "wins" : 1,
                "score" : 1400.0336900375303,
                "filename" : "20160102_164732.jpg"
            },
            {
                "matches" : 3,
                "wins" : 1,
                "score" : 1387.00607880615,
                "filename" : "20160109_151607.jpg"
            },
            {
                "matches" : 3,
                "wins" : 1,
                "score" : 1383.263693206478,
                "filename" : "20160109_152318.jpg"
            },
            {
                "matches" : 3,
                "wins" : 1,
                "score" : 1382.4634227228255,
                "filename" : "20160109_152400.jpg"
            }
        ]
    }

If you run the program again, the cached data is loaded and new match-ups can be continued using the cached data. If new photos were added, they also get added to the table data and are included in match-ups.