Adding dataset Tiny-Imagenet
Nobody has claimed this yet.
- 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
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- 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