pytorch / pytorch/vision

Adding dataset Tiny-Imagenet

Open
#6,127 4 comments 4 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

module: datasets
Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

🚀 The feature

Hello,

I would like to contribute to torchvision by providing a implementation of Tiny-Imagenet dataset.

home : https://www.kaggle.com/c/tiny-imagenet
paper : http://vision.stanford.edu/teaching/cs231n/reports/2015/pdfs/yle_project.pdf
zip : http://cs231n.stanford.edu/tiny-imagenet-200.zip

This challenge is part of Stanford Class CS 231N.
Label Classes and Bounding Boxes are provided

details:
classes : 200
image_size : 64x64x3
bbox : x0, y0, x1, y1 for each image
train split : 100 000 (500 per class)
val split : 10 000 (50 per class)
test split : 10 000 (50 per class)

Motivation, pitch

Note: the original test split doesn't have targets and bboxes.
Thus, in this implementation, I used the val split when passing train=True.

Features:

  • fast loading by creating numpy files (npy/*.npy) from the raw folder/image datasets
  • can leverage bbox

Structure:

root
├───tiny-imagenet-200.zip
├───tiny-imagenet-200
│   ├───npy <-- generated
│   │       ├───test_bboxes.npy
│   │       ├───test_data.npy
│   │       ├───test_targets.npy
│   │       ├───train_bboxes.npy
│   │       ├───train_data.npy
│   │       ├───train_targets.npy
│   ├───test
│   ├───train
│   ├───val
│   ├───words.txt
│   └───wnids.txt

Here the implementation:
https://github.com/towzeur/vision/commit/a67feb569361f440fd48ed492183de8bd8f6b585

Alternatives

No response

Additional context

No response

cc @pmeier @YosuaMichael

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Review the linked commit and the Tiny-ImageNet structure described in the issue, including the train, val, test, words.txt, wnids.txt, and generated npy files. Confirm how the proposed dataset handles targets, bounding boxes, splits, and the test set without targets; done means a reviewed torchvision implementation with those behaviors clearly defined.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
computer-vision, data
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.