Skip to content

GitLab

Commit 009a82c2 authored by Ye Liu's avatar Ye Liu Committed by Kai Chen
Browse files

Change 'modelzoo' to 'torchvision' (#1016)

parent 4e387a69
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
configs/mask_rcnn_r101_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='MaskRCNN',
pretrained='modelzoo://resnet101',
pretrained='torchvision://resnet101',
backbone=dict(
type='ResNet',
depth=101,
......
configs/mask_rcnn_r50_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='MaskRCNN',
pretrained='modelzoo://resnet50',
pretrained='torchvision://resnet50',
backbone=dict(
type='ResNet',
depth=50,
......
configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py
View file @ 009a82c2
# model settings
model = dict(
type='FasterRCNN',
pretrained='modelzoo://resnet50',
pretrained='torchvision://resnet50',
backbone=dict(
type='ResNet',
depth=50,
......
configs/retinanet_r101_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='RetinaNet',
pretrained='modelzoo://resnet101',
pretrained='torchvision://resnet101',
backbone=dict(
type='ResNet',
depth=101,
......
configs/retinanet_r50_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='RetinaNet',
pretrained='modelzoo://resnet50',
pretrained='torchvision://resnet50',
backbone=dict(
type='ResNet',
depth=50,
......
configs/rpn_r101_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='RPN',
pretrained='modelzoo://resnet101',
pretrained='torchvision://resnet101',
backbone=dict(
type='ResNet',
depth=101,
......
configs/rpn_r50_fpn_1x.py
View file @ 009a82c2
# model settings
model = dict(
type='RPN',
pretrained='modelzoo://resnet50',
pretrained='torchvision://resnet50',
backbone=dict(
type='ResNet',
depth=50,
......
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