Skip to content

Choose the better drawing to find out the best drawings in the QuickDraw dataset

Notifications You must be signed in to change notification settings

ndri/quickdrawbattle

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

QuickDrawBattle

Choose the better drawing to find out the best drawings in the QuickDraw dataset. It is running at https://andri.io/quickdrawbattle/.

Made with:

About QuickDrawBattle

Google collected over 50 million drawings with the game Quick, Draw!. They released the dataset for everyone to use, but most of the drawings are either incomplete or just bad. Let's sort out the best ones together and create a dataset of good drawings!

How does it work?

You are given 2 random drawings of a random category and you choose which is better. The drawings are then sorted based on the votes and we can see the best ones!

Since there are so many drawings, it would take a long time for even one drawing to be voted on twice and we would probably never get usable results. For that reason, there are a maximum of 25 drawings up for battle per category. They will be replaced once they have been voted on 20 times. Some categories have more than 25 drawings right now, because there used to be no limit.

The score is calculated using a lower bound of Wilson score confidence interval for a Bernoulli parameter. During score calculation, every drawing gets one invisible win, so the score of drawings with no wins is not 0 and they can be compared. The scores are also normalized to a range from 0 to 1. Table of all possible scores.

Results

You can see the results on the ranking page. To get usable data, you can GET them from /api/get_ranking. You probably want to get the key_ids from here and find the strokes from the full dataset. You can also get the strokes from here if you enable the strokes parameter, but they are in lists of coordinate pairs instead of 2 lists of x and y.

Parameters:

  • category - which category the drawings will be from
  • order - use "ascending" to get ascending order
  • strokes - use "true" to get strokes in lists of strokes which are lists of coordinate pairs
  • limit - how many drawings to return, default 25, max 1000
  • offset - how many drawings to skip
  • votemin - the minimum amount of votes for a drawing

Example requests:

Usage

  1. Clone this repo.
  2. Set up postgresql and change the database uri in app.cfg.
  3. mkdir quickdraw_dataset_bin
  4. gsutil -m cp "gs://quickdraw_dataset/full/binary/*.bin" quickdraw_dataset_bin (warning, downloads 5+ GiB of quickdrawings)
    • If you don't have gsutil, install it using curl https://sdk.cloud.google.com | bash or read here.
    • If you have the dataset somewhere else, change the dataset_dir value in config.ini.
  5. python3 -m venv quickdrawbattleenv
  6. source quickdrawbattleenv/bin/activate
  7. pip3 install -r requirements.txt
  8. python3 dbinit.py
  9. python3 app.py, serve with uwsgi and nginx (guide).

About

Choose the better drawing to find out the best drawings in the QuickDraw dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published