Skip to content

Imagenet1k prep#20

Open
PaulScotti wants to merge 4 commits into
mainfrom
imagenet1k_prep
Open

Imagenet1k prep#20
PaulScotti wants to merge 4 commits into
mainfrom
imagenet1k_prep

Conversation

@PaulScotti
Copy link
Copy Markdown
Collaborator

Changes made to get locally stored ImageNet1K to work as dataset

  • new scripts to download imagenet1k inside folder "IN1K_prep"
  • added einops scipy and h5py to requirements.txt as they were missing previously
  • added name == main to topomoe/tests/test_train.py so I can run it directly from interactive job
  • edits to topomoe/train.py such that if dataset=="folder" it will change dataloading to assume samples are coming from jpgs inside of a local folder (which contains a train/ and val/ folder where contents are subfolders of class IDs containing respective jpgs)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant