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

RecordReaderDataSetIterator: invalid output possible with some configs #4969

Closed
AlexDBlack opened this issue Apr 24, 2018 · 3 comments

Comments

@AlexDBlack
Copy link
Contributor

commented Apr 24, 2018

ImageRecordReader imageRecordReader = new ImageRecordReader(imageHeight,imageHeight,channels,new ParentPathLabelGenerator());
        imageRecordReader.initialize(new FileSplit(trainingSet),null);
        RecordReaderDataSetIterator recordReaderDataSetIterator = new RecordReaderDataSetIterator(imageRecordReader,6)

This can output data with shape [6, 126, 224, 224]

@AlexDBlack

This comment has been minimized.

Copy link
Contributor Author

commented Apr 25, 2018

Also looks like the builder can result in a similar invalid issue:
https://gist.github.com/MilanVanca/ea251702e8dbfc152a95528a7b660420

AlexDBlack added a commit that referenced this issue Apr 25, 2018
@AlexDBlack

This comment has been minimized.

Copy link
Contributor Author

commented Apr 25, 2018

@AlexDBlack AlexDBlack closed this Apr 25, 2018

@lock

This comment has been minimized.

Copy link

commented Sep 22, 2018

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.

@lock lock bot locked and limited conversation to collaborators Sep 22, 2018

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
1 participant
You can’t perform that action at this time.