diff --git a/README.md b/README.md
index f4944740..34da5e75 100644
--- a/README.md
+++ b/README.md
@@ -30,13 +30,13 @@ An easy implementation of Faster R-CNN in PyTorch.
0.7015 |
- Resnet-101 (freeze 0~4) |
+ ResNet-101 (freeze 0~4) |
~5.4 examples/sec |
~11 examples/sec |
0.7496 |
- Resnet-101 (freeze 0~5) |
+ ResNet-101 (freeze 0~5) |
~5.7 examples/sec |
~11 examples/sec |
0.7466 |
@@ -44,7 +44,7 @@ An easy implementation of Faster R-CNN in PyTorch.
- Resnet-101 (freeze 0~6)
+ ResNet-101 (freeze 0~6)
|
~7.5 examples/sec |
@@ -52,7 +52,7 @@ An easy implementation of Faster R-CNN in PyTorch.
0.7523 |
- Resnet-101 (freeze 0~7) |
+ ResNet-101 (freeze 0~7) |
~8.4 examples/sec |
~11 examples/sec |
0.6983 |
diff --git a/backbone/interface.py b/backbone/interface.py
index 631a08f2..44148fa2 100644
--- a/backbone/interface.py
+++ b/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
diff --git a/backbone/resnet101.py b/backbone/resnet101.py
index bbc0d8fb..c282b6b4 100644
--- a/backbone/resnet101.py
+++ b/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)