From 377e9bfa217b60601fb6473022970f115a5455ca Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 2 Jul 2022 15:18:52 -0700 Subject: [PATCH] Add TPU trained darknet53 weights. Add mising pretrain_cfg for some csp/darknet models. --- timm/models/cspnet.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/timm/models/cspnet.py b/timm/models/cspnet.py index 095e4701..77473052 100644 --- a/timm/models/cspnet.py +++ b/timm/models/cspnet.py @@ -45,14 +45,18 @@ default_cfgs = { 'cspresnet50w': _cfg(url=''), 'cspresnext50': _cfg( 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( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspdarknet53_ra_256-d05c7c21.pth'), 'darknet17': _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( url=''),