|
|
@ -19,7 +19,7 @@ __all__ = ['DenseNet']
|
|
|
|
|
|
|
|
|
|
|
|
def _cfg(url=''):
|
|
|
|
def _cfg(url=''):
|
|
|
|
return {
|
|
|
|
return {
|
|
|
|
'url': url, 'num_classes': 1000, 'input_size': (3, 224, 244), 'pool_size': (7, 7),
|
|
|
|
'url': url, 'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
|
|
|
|
'crop_pct': 0.875, 'interpolation': 'bicubic',
|
|
|
|
'crop_pct': 0.875, 'interpolation': 'bicubic',
|
|
|
|
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
|
|
|
|
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
|
|
|
|
'first_conv': 'features.conv0', 'classifier': 'classifier',
|
|
|
|
'first_conv': 'features.conv0', 'classifier': 'classifier',
|
|
|
|