Skip to content

Commit 5862172

Browse files
committed
Add CrossStage3 DarkNet (cs3) weights
1 parent 9be0c84 commit 5862172

File tree

1 file changed

+14
-9
lines changed

1 file changed

+14
-9
lines changed

timm/models/cspnet.py

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -57,30 +57,35 @@ def _cfg(url='', **kwargs):
5757
'sedarknet21': _cfg(url=''),
5858
'darknet53': _cfg(
5959
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/darknet53_256_c2ns-3aeff817.pth',
60-
test_input_size=(3, 288, 288), test_crop_pct=1.0, interpolation='bicubic'
60+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=1.0,
6161
),
6262
'darknetaa53': _cfg(url=''),
6363

6464
'cs3darknet_s': _cfg(
65-
url=''),
65+
url='', interpolation='bicubic'),
6666
'cs3darknet_m': _cfg(
67-
url=''),
67+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_m_c2ns-43f06604.pth',
68+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=0.95,
69+
),
6870
'cs3darknet_l': _cfg(
69-
url=''),
71+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_l_c2ns-16220c5d.pth',
72+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=0.95),
7073
'cs3darknet_x': _cfg(
7174
url=''),
7275

7376
'cs3darknet_focus_s': _cfg(
74-
url=''),
77+
url='', interpolation='bicubic'),
7578
'cs3darknet_focus_m': _cfg(
76-
url=''),
79+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_focus_m_c2ns-e23bed41.pth',
80+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=0.95),
7781
'cs3darknet_focus_l': _cfg(
78-
url=''),
82+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_focus_l_c2ns-65ef8888.pth',
83+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=0.95),
7984
'cs3darknet_focus_x': _cfg(
80-
url=''),
85+
url='', interpolation='bicubic'),
8186

8287
'cs3sedarknet_xdw': _cfg(
83-
url=''),
88+
url='', interpolation='bicubic'),
8489
}
8590

8691

0 commit comments

Comments
 (0)