Why raw_output shape=(2, 41, 41, 21)?

DrSleep / tensorflow-deeplab-resnet

DeepLab-ResNet rebuilt in TensorFlow

MIT License

1.25k stars 431 forks source link

Why raw_output shape=(2, 41, 41, 21)? #174

Closed yl3639 closed 6 years ago

yl3639 commented 6 years ago

Hi, thanks for the code sharing. I am just going through the code and I am wondering why the raw_output raw_output = net.layers['fc1_voc12']

has shape (2,41,41,21)?

Best

DrSleep commented 6 years ago

batch size, height, width, number of classes. the input image is being downsampled throughout the network until approximately (1/8) of the original resolution.

yl3639 commented 6 years ago

cool, I see! thanks!