Skip to content

Commit

Permalink
update detection configs
Browse files Browse the repository at this point in the history
  • Loading branch information
czczup committed May 31, 2022
1 parent b8857ea commit 9fc8196
Show file tree
Hide file tree
Showing 7 changed files with 10 additions and 16 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://conversationhub.blob.core.windows.net/beit-share-public/' \
# 'beit/beit_large_patch16_224_pt22k_ft22k.pth'
# pretrained = 'https://conversationhub.blob.core.windows.net/beit-share-public/beit/beit_large_patch16_224_pt22k_ft22k.pth'
pretrained = 'pretrained/beit_large_patch16_224_pt22k_ft22k.pth'
model = dict(
type='HybridTaskCascade',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://conversationhub.blob.core.windows.net/beit-share-public/' \
# 'beit/beit_large_patch16_224_pt22k_ft22k.pth'
# pretrained = 'https://conversationhub.blob.core.windows.net/beit-share-public/beit/beit_large_patch16_224_pt22k_ft22k.pth'
pretrained = 'pretrained/beit_large_patch16_224_pt22k_ft22k.pth'
model = dict(
type='HybridTaskCascadeAug',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://github.com/SwinTransformer/storage/releases/' \
# 'download/v1.0.0/swin_large_patch4_window12_384_22k.pth'
# pretrained = 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth'
pretrained = 'pretrained/swin_large_patch4_window12_384_22k.pth'
model = dict(
type='HybridTaskCascade',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://github.com/SwinTransformer/storage/releases/' \
# 'download/v1.0.0/swin_large_patch4_window12_384_22k.pth'
# pretrained = 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth'
pretrained = 'pretrained/swin_large_patch4_window12_384_22k.pth'
model = dict(
type='HybridTaskCascadeAug',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://storage.googleapis.com/vit_models/augreg/' \
# 'L_16-i21k-300ep-lr_0.001-aug_medium1-wd_0.1-do_0.1-sd_0.1--imagenet2012-steps_20k-lr_0.01-res_384.npz'
# pretrained = 'https://storage.googleapis.com/vit_models/augreg/L_16-i21k-300ep-lr_0.001-aug_medium1-wd_0.1-do_0.1-sd_0.1--imagenet2012-steps_20k-lr_0.01-res_384.npz'
pretrained = 'pretrained/L_16-i21k-300ep-lr_0.001-aug_medium1-wd_0.1-do_0.1-sd_0.1--imagenet2012-steps_20k-lr_0.01-res_384.pth'
model = dict(
backbone=dict(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,10 +29,10 @@
14, 14, None, 14, 14, None],
pretrained=pretrained),
neck=dict(
type='FPN',
in_channels=[192, 192, 192, 192],
out_channels=256,
num_outs=5))
type='FPN',
in_channels=[192, 192, 192, 192],
out_channels=256,
num_outs=5))
# optimizer
img_norm_cfg = dict(
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@
'../_base_/schedules/schedule_3x.py',
'../_base_/default_runtime.py'
]
# pretrained = 'https://github.com/czczup/ViT-Adapter/' \
# 'releases/download/v0.1.1/uniperceiver_pretrain.pth'
# pretrained = 'https://github.com/czczup/ViT-Adapter/releases/download/v0.1.1/uniperceiver_pretrain.pth'
pretrained = 'pretrained/uniperceiver_pretrain.pth'
model = dict(
backbone=dict(
Expand Down

0 comments on commit 9fc8196

Please sign in to comment.