Merge pull request CSAILVision#191 from CSAILVision/hang

Update HRNet config, model and performance

Author: hangzhaomit

README.md

@@ -52,7 +52,7 @@ Encoder:
 - ResNet18/ResNet18dilated
 - ResNet50/ResNet50dilated
 - ResNet101/ResNet101dilated
-- HRNet (HRNetV2-W48)
+- HRNetV2 (W48)
 
 Decoder:
 - C1 (one convolution module)
@@ -144,12 +144,12 @@ IMPORTANT: The base ResNet in our repository is a customized (different from the
         <td>2.3</td>
     </tr>
     <tr>
-        <td rowspan="2">HRNetV2-W48</td>
-        <td>No</td><td>41.74</td><td>80.59</td><td>61.17</td>
+        <td rowspan="2">HRNetV2</td>
+        <td>No</td><td>42.03</td><td>80.77</td><td>61.40</td>
         <td>5.8</td>
     </tr>
     <tr>
-        <td>Yes</td><td>42.99</td><td>81.25</td><td>62.12</td>
+        <td>Yes</td><td>43.20</td><td>81.47</td><td>62.34</td>
         <td>1.9</td>
     </tr>
 

config/ade20k-hrnetv2.yaml

@@ -16,7 +16,7 @@ MODEL:
 
 TRAIN:
   batch_size_per_gpu: 2
-  num_epoch: 20
+  num_epoch: 30
   start_epoch: 0
   epoch_iters: 5000
   optim: "SGD"
@@ -33,10 +33,10 @@ TRAIN:
 
 VAL:
   visualize: False
-  checkpoint: "epoch_20.pth"
+  checkpoint: "epoch_30.pth"
 
 TEST:
-  checkpoint: "epoch_20.pth"
+  checkpoint: "epoch_30.pth"
   result: "./"
 
 DIR: "ckpt/ade20k-hrnetv2-c1"