AI-MXNet-Gluon-ModelZoo

 view release on metacpan or  search on metacpan

MANIFEST  view on Meta::CPAN

lib/AI/MXNet/Gluon/ModelZoo/Vision/MobileNet.pm
lib/AI/MXNet/Gluon/ModelZoo/Vision/ResNet.pm
lib/AI/MXNet/Gluon/ModelZoo/Vision/SqueezeNet.pm
lib/AI/MXNet/Gluon/ModelZoo/Vision/VGG.pm
Makefile.PL
MANIFEST
META.json
META.yml
README
t/AI-MXNet-Gluon-ModelZoo.t
t/test_gluon_model_zoo.t

META.json  view on Meta::CPAN

            "ExtUtils::MakeMaker" : "6.30"
         }
      },
      "runtime" : {
         "requires" : {
            "AI::MXNet" : "1.31",
            "AI::MXNet::Gluon::Contrib" : "1.3",
            "IO::Uncompress::Unzip" : "0"
         }
      },
      "test" : {
         "requires" : {}
      }
   },
   "release_status" : "stable",
   "version" : "1.33"
}

Makefile.PL  view on Meta::CPAN

  "EXE_FILES" => [],
  "LICENSE" => "apache_2_0",
  "NAME" => "AI::MXNet::Gluon::ModelZoo",
  "PREREQ_PM" => {
    "AI::MXNet" => "1.31",
    "AI::MXNet::Gluon::Contrib" => "1.3",
    "IO::Uncompress::Unzip" => "0"
  },
  "TEST_REQUIRES" => {},
  "VERSION" => "1.33",
  "test" => {
    "TESTS" => "t/*.t"
  }
);


my %FallbackPrereqs = (
  "AI::MXNet" => "1.31",
  "AI::MXNet::Gluon::Contrib" => "1.3",
  "IO::Uncompress::Unzip" => "0"
);

t/AI-MXNet-Gluon-ModelZoo.t  view on Meta::CPAN

#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

use strict;
use warnings;
use Test::More tests => 1;
BEGIN { use_ok('AI::MXNet::Gluon::ModelZoo') };

t/test_gluon_model_zoo.t  view on Meta::CPAN

# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

use strict;
use warnings;
use AI::MXNet::Gluon::ModelZoo qw(get_model);
use Test::More tests => 34;

sub test_models
{
    my @all_models = ('resnet34_v1', 'resnet18_v1', 'resnet50_v1', 'resnet101_v1', 'resnet152_v1',
                  'resnet18_v2', 'resnet34_v2', 'resnet50_v2', 'resnet101_v2', 'resnet152_v2',
                  'vgg11', 'vgg13', 'vgg16', 'vgg19',
                  'vgg11_bn', 'vgg13_bn', 'vgg16_bn', 'vgg19_bn',
                  'alexnet', 'inceptionv3',
                  'densenet121', 'densenet161', 'densenet169', 'densenet201',
                  'squeezenet1.0', 'squeezenet1.1',
                  'mobilenet1.0', 'mobilenet0.75', 'mobilenet0.5', 'mobilenet0.25',
                  'mobilenetv2_1.0', 'mobilenetv2_0.75', 'mobilenetv2_0.5', 'mobilenetv2_0.25');
    my %pretrained_to_test = ('squeezenet1.1' => 1);

    for my $model_name (@all_models)
    {
            my $test_pretrain = exists $pretrained_to_test{ $model_name };
            my $model = get_model($model_name, pretrained=>$test_pretrain, root=>'model/');
            my $data_shape = $model_name !~ /inception/ ? [2, 3, 224, 224] : [2, 3, 299, 299];
            if(not $test_pretrain)
            {
                $model->collect_params()->initialize();
            }
            $model->hybridize();
            $model->(mx->nd->random->uniform(shape=>$data_shape))->wait_to_read;
            ok(1, "forward for $model_name");
    }
}

test_models();