DeepRegNet /
DeepReg
| @@ 89-109 (lines=21) @@ | ||
| 86 | output = network.call(inputs) |
|
| 87 | assert inputs.shape == output.shape |
|
| 88 | ||
| 89 | def test_get_config(self): |
|
| 90 | config = dict( |
|
| 91 | image_size=(4, 5, 6), |
|
| 92 | out_channels=3, |
|
| 93 | num_channel_initial=2, |
|
| 94 | depth=2, |
|
| 95 | extract_levels=(0, 1), |
|
| 96 | out_kernel_initializer="he_normal", |
|
| 97 | out_activation="softmax", |
|
| 98 | pooling=False, |
|
| 99 | concat_skip=False, |
|
| 100 | use_additive_upsampling=True, |
|
| 101 | encode_kernel_sizes=[7, 3, 3], |
|
| 102 | decode_kernel_sizes=3, |
|
| 103 | strides=2, |
|
| 104 | padding="same", |
|
| 105 | name="Test", |
|
| 106 | ) |
|
| 107 | network = LocalNet(**config) |
|
| 108 | got = network.get_config() |
|
| 109 | assert got == config |
|
| 110 | ||
| @@ 53-72 (lines=20) @@ | ||
| 50 | output = network.call(inputs) |
|
| 51 | assert inputs.shape == output.shape |
|
| 52 | ||
| 53 | def test_get_config(self): |
|
| 54 | config = dict( |
|
| 55 | image_size=(4, 5, 6), |
|
| 56 | out_channels=3, |
|
| 57 | num_channel_initial=2, |
|
| 58 | depth=2, |
|
| 59 | extract_levels=(0, 1), |
|
| 60 | out_kernel_initializer="he_normal", |
|
| 61 | out_activation="softmax", |
|
| 62 | pooling=False, |
|
| 63 | concat_skip=False, |
|
| 64 | encode_kernel_sizes=3, |
|
| 65 | decode_kernel_sizes=3, |
|
| 66 | strides=2, |
|
| 67 | padding="same", |
|
| 68 | name="Test", |
|
| 69 | ) |
|
| 70 | network = UNet(**config) |
|
| 71 | got = network.get_config() |
|
| 72 | assert got == config |
|
| 73 | ||
| @@ 93-112 (lines=20) @@ | ||
| 90 | ) |
|
| 91 | assert "GlobalNet requires `depth` or `extract_levels`" in str(err_info.value) |
|
| 92 | ||
| 93 | def test_get_config(self): |
|
| 94 | config = dict( |
|
| 95 | image_size=(4, 5, 6), |
|
| 96 | out_channels=3, |
|
| 97 | num_channel_initial=2, |
|
| 98 | depth=2, |
|
| 99 | extract_levels=(2,), |
|
| 100 | out_kernel_initializer="he_normal", |
|
| 101 | out_activation="softmax", |
|
| 102 | pooling=False, |
|
| 103 | concat_skip=False, |
|
| 104 | encode_kernel_sizes=[7, 3, 3], |
|
| 105 | decode_kernel_sizes=3, |
|
| 106 | strides=2, |
|
| 107 | padding="same", |
|
| 108 | name="Test", |
|
| 109 | ) |
|
| 110 | network = GlobalNet(**config) |
|
| 111 | got = network.get_config() |
|
| 112 | assert got == config |
|
| 113 | ||
