Ejemplo n.º 1
0
    def test_inceptionv3_2(self):

        if self.data_dir is None:
            unittest.TestCase.skipTest(self, "DLPY_DATA_DIR is not set in the environment variables")
        file_dependency = self.data_dir + 'InceptionV3_weights.kerasmodel.h5'
        if not file_exist_on_server(self.s, file_dependency):
            unittest.TestCase.skipTest(self, "File, {}, not found.".format(file_dependency))

        from dlpy.applications import InceptionV3
        model = InceptionV3(self.s,
                            model_table='INCEPTIONV3',
                            n_classes=1000,
                            n_channels=3,
                            width=299,
                            height=299,
                            scale=1,
                            offsets=(103.939, 116.779, 123.68),
                            pre_trained_weights=True,
                            pre_trained_weights_file=self.data_dir + 'InceptionV3_weights.kerasmodel.h5',
                            include_top=True)
        model.print_summary()

        # test random_crop and mutation
        model2 = InceptionV3(self.s, model_table='VGG16', n_classes=1000, n_channels=3,
                             width=299, height=299, scale=1,
                             offsets=None,
                             random_crop='unique',
                             random_flip='hv',
                             random_mutation='random',
                             pre_trained_weights=True,
                             pre_trained_weights_file=self.data_dir + 'InceptionV3_weights.kerasmodel.h5',
                             include_top=True)
        model2.print_summary()
Ejemplo n.º 2
0
    def test_inceptionv3_3(self):

        if self.data_dir is None:
            unittest.TestCase.skipTest(self, "DLPY_DATA_DIR is not set in the environment variables")

        from dlpy.applications import InceptionV3
        model = InceptionV3(self.s,
                            model_table='INCEPTIONV3',
                            n_classes=3,
                            n_channels=3,
                            width=299,
                            height=299,
                            scale=1,
                            offsets=(103.939, 116.779, 123.68),
                            pre_trained_weights=True,
                            pre_trained_weights_file=self.data_dir+'InceptionV3_weights.kerasmodel.h5',
                            include_top=False)
        model.print_summary()
Ejemplo n.º 3
0
 def test_inceptionv3_1(self):
     from dlpy.applications import InceptionV3
     model = InceptionV3(self.s, n_classes=3)
     model.print_summary()
Ejemplo n.º 4
0
 def test_inceptionv3(self):
     from dlpy.applications import InceptionV3
     model = InceptionV3(self.s)
     model.print_summary()