def __init__(self, input_channels):
super(SingleImageBreastModel, self).__init__()
self.view_resnet = resnet22(input_channels)
self.fc1 = nn.Linear(256, 256)
self.output_layer = layers.OutputLayer(256, (2, 2))
self.all_views_avg_pool = layers.AllViewsAvgPool()
self.all_views_gaussian_noise_layer = layers.AllViewsGaussianNoise(
0.01)
def __init__(self, input_channels):
super(SplitBreastModel, self).__init__()
self.four_view_resnet = FourViewResNet(input_channels)
self.fc1_cc = nn.Linear(256 * 2, 256 * 2)
self.fc1_mlo = nn.Linear(256 * 2, 256 * 2)
self.output_layer_cc = layers.OutputLayer(256 * 2, (4, 2))
self.output_layer_mlo = layers.OutputLayer(256 * 2, (4, 2))
self.all_views_avg_pool = layers.AllViewsAvgPool()
self.all_views_pad = layers.AllViewsPad()
self.all_views_gaussian_noise_layer = layers.AllViewsGaussianNoise(0.01)
def __init__(self, input_channels):
super(ImageBreastModel, self).__init__()
self.four_view_resnet = FourViewResNet(input_channels)
self.fc1_lcc = nn.Linear(256, 256)
self.fc1_rcc = nn.Linear(256, 256)
self.fc1_lmlo = nn.Linear(256, 256)
self.fc1_rmlo = nn.Linear(256, 256)
self.output_layer_lcc = layers.OutputLayer(256, (4, 2))
self.output_layer_rcc = layers.OutputLayer(256, (4, 2))
self.output_layer_lmlo = layers.OutputLayer(256, (4, 2))
self.output_layer_rmlo = layers.OutputLayer(256, (4, 2))
self.all_views_avg_pool = layers.AllViewsAvgPool()
self.all_views_gaussian_noise_layer = layers.AllViewsGaussianNoise(
0.01)