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

some renaming

parent e23ee232
......@@ -3,7 +3,7 @@ model = dict(
type='RetinaNet',
pretrained='modelzoo://resnet50',
backbone=dict(
type='resnet',
type='ResNet',
depth=50,
num_stages=4,
out_indices=(0, 1, 2, 3),
......
......@@ -5,7 +5,7 @@ from .rpn import RPN
from .fast_rcnn import FastRCNN
from .faster_rcnn import FasterRCNN
from .mask_rcnn import MaskRCNN
from .retina_net import RetinaNet
from .retinanet import RetinaNet
__all__ = [
'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