Skip to content

modestyachts/ImageNetV2_pytorch

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

Pytorch Dataloader for ImageNet V2

First Install

pip install git+https://github.com/modestyachts/ImageNetV2_pytorch

Usage:

from imagenetv2_pytorch import ImageNetV2Dataset
from torch.utils.data import DataLoader

dataset = ImageNetV2Dataset("matched-frequency") # supports matched-frequency, threshold-0.7, top-images variants
dataloader = torch.utils.data.DataLoader(dataset, batch_size=32) # use whatever batch size you wish
# feed into pytorch code

About

ImageNetV2 Pytorch Dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages