fix a bug in deeplabv3 (PaddlePaddle#541)

* fix a shape typo * replace F.droput with nn.Dropout * add ann 80k configs * add ann 80k configs * add ann 80k * add gcnet 80k * add fastscnn 160k * take align_corners as a configurable argument * delete ann 160k * delete ann 160k * delete gcnet 160k * fix a bug
shijun18 · Nov 2, 2020 · 9ec92a1 · 9ec92a1
1 parent 062cebf
commit 9ec92a1
Show file tree

Hide file tree

Showing 7 changed files with 1 addition and 59 deletions.
diff --git a/dygraph/configs/ann/ann_resnet101_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/ann/ann_resnet101_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/configs/ann/ann_resnet50_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/ann/ann_resnet50_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/configs/deeplabv3/deeplabv3_resnet101_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/deeplabv3/deeplabv3_resnet101_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/configs/deeplabv3/deeplabv3_resnet50_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/deeplabv3/deeplabv3_resnet50_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/configs/gcnet/gcnet_resnet101_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/gcnet/gcnet_resnet101_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/configs/gcnet/gcnet_resnet50_os8_cityscapes_1024x512_160k.yml b/dygraph/configs/gcnet/gcnet_resnet50_os8_cityscapes_1024x512_160k.yml
diff --git a/dygraph/paddleseg/models/deeplab.py b/dygraph/paddleseg/models/deeplab.py
@@ -161,7 +161,7 @@ def __init__(self,
 
         self.head = DeepLabV3Head(num_classes, backbone_indices,
                                   backbone_channels, aspp_ratios,
-                                  aspp_out_channels)
+                                  aspp_out_channels, align_corners)
         self.align_corners = align_corners
         self.pretrained = pretrained
         self.init_weight()