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)
def test_keypoints_layer2(self): if not __dev__: with self.assertRaises(DLPyError): Keypoints(not_a_parameter=1)
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))