Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Update Image training code for downstream datasets (iWildcam, transfer datasets) #2310

Open
wants to merge 6 commits into
base: main
Choose a base branch
from

Conversation

ohaijen
Copy link
Contributor

@ohaijen ohaijen commented May 29, 2024

Update trochvision/train.py to accomodate:

  • iWildcam. This code would be read through the ImageFolder format, but requires different resizes and crops.
  • Transfer datasets (Aircraft, DTD, Flowers) - these would be read through the TorchVision API and cached in /tmp. These don't need any new transforms, just the class loading.

@ohaijen ohaijen requested a review from anmarques May 30, 2024 15:44
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant