Add TPU trained darknet53 weights. Add mising pretrain_cfg for some csp/darknet models.

pull/1327/head
Ross Wightman 2 years ago
parent c170ba3173
commit 377e9bfa21

@ -45,14 +45,18 @@ default_cfgs = {
'cspresnet50w': _cfg(url=''), 'cspresnet50w': _cfg(url=''),
'cspresnext50': _cfg( 'cspresnext50': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspresnext50_ra_224-648b4713.pth', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspresnext50_ra_224-648b4713.pth',
input_size=(3, 224, 224), pool_size=(7, 7), crop_pct=0.875 # FIXME I trained this at 224x224, not 256 like ref impl
), ),
'cspdarknet53': _cfg( 'cspdarknet53': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspdarknet53_ra_256-d05c7c21.pth'), url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspdarknet53_ra_256-d05c7c21.pth'),
'darknet17': _cfg(url=''), 'darknet17': _cfg(url=''),
'darknet21': _cfg(url=''), 'darknet21': _cfg(url=''),
'darknet53': _cfg(url=''), 'sedarknet21': _cfg(url=''),
'darknet53': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/darknet53_256_c2ns-3aeff817.pth',
test_input_size=(3, 288, 288), test_crop_pct=1.0, interpolation='bicubic'
),
'darknetaa53': _cfg(url=''),
'cs2darknet_m': _cfg( 'cs2darknet_m': _cfg(
url=''), url=''),

Loading…
Cancel
Save