search

AstarLight / Satellite-Segmentation

615 stars 233 forks source link

关于segnet的一点问题及正确思路 #71

Open wanghao15536870732 opened 3 years ago

wanghao15536870732 commented 3 years ago

segnet模型是需要在pooling时记住pooling index的，然后在upsampling的时候借用pooling index去恢复图像，keras是在upsampling的时候是没有根据pooling index去特定恢复的功能，这里需要自己用tf去写一个pooling与upsampling。

具体可参考#57