Skip to content

JanRuettinger/dog_heads_dataset

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dog Dataset

The dataset consits of ~22k images of predominately front facing dog heads with a size of 256x256 pixels. It was derived from the Tsinghua Dog Dataset.

How can I get the dataset?

  1. Download the Tsinghua Dog Dataset here.
  2. Install libs for preprocessing: pip install -r requirements.txt
  3. Switch to scripts folder: cd scripts
  4. Run preprocessing steps 1-3: python step_1.py etc.
  5. Alternatively you can start a jupyter notebook and run the provided notebooks: jupyter-lab.

Preprocessing details

  • During step 1 images are cropped according to bounding boxes provided by Tsinghua Dogs Dataset creators.
  • During step 2 a second dog face detector is used to extract more accurate bounding boxes and key frames (eyes, nose, forehead). Both are used to refine the final cropped image. In addition, small images and images showing a dog from the side are filtered out.
  • During step 3 the dataset is split into a train and validation set.

Example images

image_1 image_2 image_3 image_4 image_5

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages