From 4042a94f8fe4651fb944b62361bd8a0f5680aaed Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 26 Jul 2022 16:36:04 -0700 Subject: [PATCH] Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks. --- timm/models/cspnet.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/timm/models/cspnet.py b/timm/models/cspnet.py index 93cb93ee..2c09e7e3 100644 --- a/timm/models/cspnet.py +++ b/timm/models/cspnet.py @@ -97,9 +97,11 @@ default_cfgs = { url='', interpolation='bicubic'), 'cs3edgenet_x': _cfg( - url='', interpolation='bicubic'), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3edgenet_x_c2-2e1610a9.pth', + interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=1.0), 'cs3se_edgenet_x': _cfg( - url='', interpolation='bicubic'), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3se_edgenet_x_c2ns-76f8e3ac.pth', + interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0), }