Skip to content
Snippets Groups Projects
Commit cae36ab6 authored by Kai Chen's avatar Kai Chen
Browse files

some renaming

parent e23ee232
No related branches found
No related tags found
No related merge requests found
...@@ -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',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment