Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix transform call in ImageFolderDataset #15672

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Fix transform call in ImageFolderDataset #15672

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
55831be
Fix transform call in ImageFolderDataset
john- Jul 27, 2019
1be9f4c
Merge branch 'master' into fix_image_folder_dataset
john- Jul 30, 2019
644540e
Trigger notification
john- Aug 19, 2019
5cfe4e6
trigger CI
ThomasDelteil Sep 1, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions perl-package/AI-MXNet/lib/AI/MXNet/Gluon/Data/Vision.pm
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ use Mouse;
Whether to load the training or testing set.
Defaults to True
transform : function
A user defined callback that transforms each instance. For example
A user defined callback that transforms each instance. For example:

transform => sub { my ($data, $label) = @_; return ($data->astype('float32')/255, $label) }
=cut
Expand Down Expand Up @@ -367,7 +367,7 @@ extends 'AI::MXNet::Gluon::Data::Set';
=cut
has 'root' => (is => 'rw', isa => 'Str');
has 'flag' => (is => 'rw', isa => 'Bool', default => 1);
has 'transform' => (is => 'rw', isa => 'Maybe[CodeRef]');
has 'transform' => (is => 'ro', isa => 'Maybe[CodeRef]');
has [qw/exts
synsets
items/] => (is => 'rw', init_arg => undef);
Expand Down Expand Up @@ -421,7 +421,7 @@ method at(Int $idx)
my $label = $self->items->[$idx][1];
if(defined $self->transform)
{
return [$self->transform($img, $label)];
return [$self->transform->($img, $label)];
}
return [$img, $label];
}
Expand Down