Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

add dataflow module #1028

Draft
wants to merge 11 commits into
base: master
Choose a base branch
from
Draft

add dataflow module #1028

wants to merge 11 commits into from

Conversation

1FengL
Copy link
Contributor

@1FengL 1FengL commented Jul 25, 2019

Checklist

  • I've tested that my changes are compatible with the latest version of Tensorflow.
  • I've read the Contribution Guidelines
  • I've updated the documentation if necessary.

Motivation and Context

Create tensorlayer.dataflow modeule for data loading, augmentation, preprocessing, multiprocessing, shuffle and batch.

Description

TODOs:

  • documentation
  • readme
  • lazy import
  • example: train ResNet-50 on ILSVRC
  • better design for TransformedDataset
  • better design for AugmentedDataset

CIFAR10_URL = 'https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz'


class CIFAR10(Dataset):
Copy link
Member

@zsdonghao zsdonghao Jul 25, 2019

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

move to tl.files?? any idea? use tl.files.load_dataset_cifar10 ??

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think those data-related functions originally in tl.files should be moved to tl.data in the future so that all data-related functionalities are in the same place.

urlretrieve(url_source, filepath, reporthook=_dlProgress)


def maybe_download_and_extract(filename, working_directory, url_source, extract=False, expected_bytes=None):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why copy the code ..

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.

3 participants