def expanding_block(self, node, current_level, is_training):
for i in range(self.repeats):
node = self.conv(node, current_level, str(i), is_training)
if i < self.repeats and self.dropout_ratio > 0:
node = dropout(node, self.dropout_ratio,
'drop' + str(current_level), is_training)
return node
def expanding_block(self, node, current_level, is_training):
node = self.conv(node, current_level, '0', is_training)
node = dropout(node, 0.5, 'drop' + str(current_level), is_training)
node = self.conv(node, current_level, '1', is_training)
return node
def contracting_block(self, node, current_level, is_training):
node = self.conv(node, current_level, '_0', is_training)
node = dropout(node, 0.5, 'drop' + str(current_level), is_training)
node = self.conv(node, current_level, '_1', is_training)
return node
def expanding_block(self, node, current_level, is_training):
node = self.conv(node, current_level, '', is_training)
return dropout(node,
self.dropout_rate,
name='dropout',
is_training=is_training)