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)