lsh1994 / keras-segmentation

Get started with Semantic Segmentation based on Keras, including FCN32/FCN8/SegNet/U-Net
166 stars 55 forks source link

add utils tensorflow CPU version #15

Closed kongnuonuo closed 5 years ago

kongnuonuo commented 5 years ago

The script can run under tensorflow(CPU version). It parsed max pooling indices strictly using formula ( (b height + y) width + x) * channels + c. For more details, please refer to https://www.tensorflow.org/api_docs/python/tf/nn/max_pool_with_argmax