Merge pull request #3 from A01781042/sibaja

Sibaja

configs/_base_/models/deeplabv3_r50-d8.py

@@ -3,8 +3,8 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 data_preprocessor = dict(
     type='SegDataPreProcessor',
-    mean=[123.675, 116.28, 103.53],
-    std=[58.395, 57.12, 57.375],
+    mean=[0.4082, 0.3791, 0.2815],
+    std=[0.1351, 0.1022, 0.0931],
     bgr_to_rgb=True,
     pad_val=0,
     seg_pad_val=255)

configs/_base_/models/deeplabv3plus_r50-d8_deepglobe.py

@@ -0,0 +1,54 @@
+# model settings
+norm_cfg = dict(type='SyncBN', requires_grad=True)
+data_preprocessor = dict(
+    type='SegDataPreProcessor',
+    mean=[0.4082, 0.3791, 0.2815],
+    std=[0.1351, 0.1022, 0.0931],
+    bgr_to_rgb=True,
+    pad_val=0,
+    seg_pad_val=255)
+model = dict(
+    type='EncoderDecoder',
+    data_preprocessor=data_preprocessor,
+    pretrained='open-mmlab://resnet50_v1c',
+    backbone=dict(
+        type='ResNetV1c',
+        depth=50,
+        num_stages=4,
+        out_indices=(0, 1, 2, 3),
+        dilations=(1, 1, 2, 4),
+        strides=(1, 2, 1, 1),
+        norm_cfg=norm_cfg,
+        norm_eval=False,
+        style='pytorch',
+        contract_dilation=True),
+    decode_head=dict(
+        type='DepthwiseSeparableASPPHead',
+        in_channels=2048,
+        in_index=3,
+        channels=512,
+        dilations=(1, 12, 24, 36),
+        c1_in_channels=256,
+        c1_channels=48,
+        dropout_ratio=0.1,
+        num_classes=7,
+        norm_cfg=norm_cfg,
+        align_corners=False,
+        loss_decode=dict(
+            type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0)),
+    auxiliary_head=dict(
+        type='FCNHead',
+        in_channels=1024,
+        in_index=2,
+        channels=256,
+        num_convs=1,
+        concat_input=False,
+        dropout_ratio=0.1,
+        num_classes=7,
+        norm_cfg=norm_cfg,
+        align_corners=False,
+        loss_decode=dict(
+            type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
+    # model training and testing settings
+    train_cfg=dict(),
+    test_cfg=dict(mode='whole'))

configs/_base_/models/pspnet_unet_deepglobe_s5-d16.py

@@ -2,8 +2,8 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 data_preprocessor = dict(
     type='SegDataPreProcessor',
-    mean=[123.675, 116.28, 103.53],
-    std=[58.395, 57.12, 57.375],
+    mean=[0.4082, 0.3791, 0.2815],
+    std=[0.1351, 0.1022, 0.0931],
     bgr_to_rgb=True,
     pad_val=0,
     seg_pad_val=255)
@@ -35,14 +35,14 @@
         channels=16,
         pool_scales=(1, 2, 3, 6),
         dropout_ratio=0.1,
-        num_classes=2,
+        num_classes=7,
         norm_cfg=norm_cfg,
         align_corners=False,
         loss_decode=dict(
             type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0)),
     auxiliary_head=dict(
         type='FCNHead',
-        in_channels=1024,
+        in_channels=128,
         in_index=3,
         channels=256,
         num_convs=1,

configs/deeplabv3plus/deeplabv3plus_r50-d8_4xb2-40k_deepglobe-512x1024.py

@@ -0,0 +1,8 @@
+_base_ = [
+    '../_base_/models/deeplabv3plus_r50-d8_deepglobe.py',
+    '../_base_/datasets/deepglobe.py', '../_base_/default_runtime.py',
+    '../_base_/schedules/schedule_40k.py'
+]
+crop_size = (256, 256)
+data_preprocessor = dict(size=crop_size)
+model = dict(data_preprocessor=data_preprocessor)