EAST Text Detection Network: Implementation in TensorFlow and Keras
The code defines a class called 'EAST', which creates a neural network for detecting text in images. This network is based on the VGG16 architecture and employs a combination of convolutional and upsampling layers to extract features from the input image. The network's output consists of scores and coordinates that pinpoint potential text regions within the image.
The 'quad_loss' function acts as the loss function during training. It calculates the loss for each output score and coordinate using a combination of binary cross-entropy and smooth L1 loss. The loss is weighted based on the importance of each output.
Overall, the code provides an implementation of the EAST text detection network and its accompanying loss function. This can be used for training and testing on text detection datasets.
原文地址: https://www.cveoy.top/t/topic/oBh8 著作权归作者所有。请勿转载和采集!