Mastering Computer Vision with TensorFlow 2.x

Training an hourglass network

The hourglass network has already been described. In this section, we will explain the code behind training the network.

If you want to train your own hourglass network, follow the instructions at https://github.com/yuanyuanli85/Stacked_Hourglass_Network_Keras.

The code for training an hourglass network is as follows:

def build_model(self, mobile=False, show=False):if mobile:   self.model = create_hourglass_network(self.num_classes, self.num_stacks, self.num_channels, self.inres, self.outres, bottleneck_mobile) else:    self.model = create_hourglass_network(self.num_classes, self.num_stacks,self.num_channels, self.inres, self.outres, bottleneck_block)# show model summary and layer name  if show: self.model.summary(def ...