Example #1
0
 def test_keypoints_layer1(self):
     dict1 = Keypoints(name='keypoints',
                       n=100,
                       src_layers=[Dense(name='fc',
                                         n=100)]).to_model_params()
     self.assertTrue(self.sample_syntax['keypoints1'] == dict1)
Example #2
0
 def test_keypoints_layer2(self):
     if not __dev__:
         with self.assertRaises(DLPyError):
             Keypoints(not_a_parameter=1)
Example #3
0
 def test_model16(self):
     model = Sequential(self.s, model_table='Simple_CNN')
     model.add(layer=InputLayer(n_channels=1, height=10, width=10))
     model.add(layer=Keypoints(n=10))
     self.assertTrue(model.summary.loc[1, 'Number of Parameters'] == (1000, 10))