Rename

README.md

@@ -30,29 +30,29 @@ An easy implementation of Faster R-CNN in PyTorch.
         <td>0.7015</td>
     </tr>
     <tr style="color: gray;">
-        <td>Resnet-101 (freeze 0~4)</td>
+        <td>ResNet-101 (freeze 0~4)</td>
         <td>~5.4 examples/sec</td>
         <td>~11 examples/sec</td>
         <td>0.7496</td>
     </tr>
     <tr style="color: gray;">
-        <td>Resnet-101 (freeze 0~5)</td>
+        <td>ResNet-101 (freeze 0~5)</td>
         <td>~5.7 examples/sec</td>
         <td>~11 examples/sec</td>
         <td>0.7466</td>
     </tr>
     <tr>
         <td>
             <a href="https://drive.google.com/open?id=1t-lv8bNcPnoeVru3PD3BLLMmFHsUY8u9">
-                Resnet-101 (freeze 0~6)
+                ResNet-101 (freeze 0~6)
             </a>
         </td>
         <td>~7.5 examples/sec</td>
         <td>~11 examples/sec</td>
         <td>0.7523</td>
     </tr>
     <tr style="color: gray;">
-        <td>Resnet-101 (freeze 0~7)</td>
+        <td>ResNet-101 (freeze 0~7)</td>
         <td>~8.4 examples/sec</td>
         <td>~11 examples/sec</td>
         <td>0.6983</td>

backbone/interface.py

@@ -9,8 +9,8 @@ def from_name(name: str) -> Type['Interface']:
             from backbone.vgg16 import Vgg16
             return Vgg16
         elif name == 'resnet101':
-            from backbone.resnet101 import Resnet101
-            return Resnet101
+            from backbone.resnet101 import ResNet101
+            return ResNet101
         else:
             raise ValueError
 

backbone/resnet101.py

@@ -4,7 +4,7 @@
 from backbone.interface import Interface
 
 
-class Resnet101(Interface):
+class ResNet101(Interface):
 
     def __init__(self, pretrained: bool):
         super().__init__(pretrained)