Commit cae36ab6 authored by Kai Chen's avatar Kai Chen
Browse files

some renaming

parent e23ee232
...@@ -3,7 +3,7 @@ model = dict( ...@@ -3,7 +3,7 @@ model = dict(
type='RetinaNet', type='RetinaNet',
pretrained='modelzoo://resnet50', pretrained='modelzoo://resnet50',
backbone=dict( backbone=dict(
type='resnet', type='ResNet',
depth=50, depth=50,
num_stages=4, num_stages=4,
out_indices=(0, 1, 2, 3), out_indices=(0, 1, 2, 3),
......
...@@ -5,7 +5,7 @@ from .rpn import RPN ...@@ -5,7 +5,7 @@ from .rpn import RPN
from .fast_rcnn import FastRCNN from .fast_rcnn import FastRCNN
from .faster_rcnn import FasterRCNN from .faster_rcnn import FasterRCNN
from .mask_rcnn import MaskRCNN from .mask_rcnn import MaskRCNN
from .retina_net import RetinaNet from .retinanet import RetinaNet
__all__ = [ __all__ = [
'BaseDetector', 'SingleStageDetector', 'TwoStageDetector', 'RPN', 'BaseDetector', 'SingleStageDetector', 'TwoStageDetector', 'RPN',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment