-
Notifications
You must be signed in to change notification settings - Fork 39
/
densenet.py
45 lines (34 loc) · 904 Bytes
/
densenet.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author: zcy
# @Date: 2019-02-15 13:02:27
# @Last Modified by: zcy
# @Last Modified time: 2019-02-15 15:36:26
from models.StereoCNN.Densenet_module import *
__all__ = [
'densenet121_3d', 'densenet169_3d', 'densenet201_3d', 'densenet264_3d'
]
def densenet121_3d(**kwargs):
model = DenseNet(
growth_rate=32,
block_config=(6, 12, 24, 16),
**kwargs)
return model
def densenet169_3d(**kwargs):
model = DenseNet(
growth_rate=32,
block_config=(6, 12, 32, 32),
**kwargs)
return model
def densenet201_3d(**kwargs):
model = DenseNet(
growth_rate=32,
block_config=(6, 12, 48, 32),
**kwargs)
return model
def densenet264_3d(**kwargs):
model = DenseNet(
growth_rate=32,
block_config=(6, 12, 64, 48),
**kwargs)
return model