update readme

xvjiarui · Dec 20, 2018 · 21467a7 · 21467a7
1 parent ba3543b
commit 21467a7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -58,14 +58,14 @@ Results and models are available in the [Model zoo](MODEL_ZOO.md).
 
 |                    | ResNet   | ResNeXt  | SENet    | VGG      |
 |--------------------|:--------:|:--------:|:--------:|:--------:|
-| RPN                | ✓        | ☐        | ☐        | ✗        |
-| Fast R-CNN         | ✓        | ☐        | ☐        | ✗        |
-| Faster R-CNN       | ✓        | ☐        | ☐        | ✗        |
-| Mask R-CNN         | ✓        | ☐        | ☐        | ✗        |
-| Cascade R-CNN      | ✓        | ☐        | ☐        | ✗        |
-| Cascade Mask R-CNN | ✓        | ☐        | ☐        | ✗        |
+| RPN                | ✓        | ✓        | ☐        | ✗        |
+| Fast R-CNN         | ✓        | ✓        | ☐        | ✗        |
+| Faster R-CNN       | ✓        | ✓        | ☐        | ✗        |
+| Mask R-CNN         | ✓        | ✓        | ☐        | ✗        |
+| Cascade R-CNN      | ✓        | ✓        | ☐        | ✗        |
+| Cascade Mask R-CNN | ✓        | ✓        | ☐        | ✗        |
 | SSD                | ✗        | ✗        | ✗        | ☐        |
-| RetinaNet          | ✓        | ☐        | ☐        | ✗        |
+| RetinaNet          | ✓        | ✓        | ☐        | ✗        |
 
 ## Installation
 

diff --git a/mmdet/models/backbones/resnet.py b/mmdet/models/backbones/resnet.py
@@ -86,7 +86,11 @@ def __init__(self,
             self.conv1_stride = stride
             self.conv2_stride = 1
         self.conv1 = nn.Conv2d(
-            inplanes, planes, kernel_size=1, stride=self.conv1_stride, bias=False)
+            inplanes,
+            planes,
+            kernel_size=1,
+            stride=self.conv1_stride,
+            bias=False)
         self.conv2 = nn.Conv2d(
             planes,
             planes,

diff --git a/mmdet/models/backbones/resnext.py b/mmdet/models/backbones/resnext.py
@@ -1,7 +1,6 @@
 import math
 
 import torch.nn as nn
-import torch.utils.checkpoint as cp
 
 from .resnet import ResNet
 from .resnet import Bottleneck as _Bottleneck