Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
stoneliu1981
pytorch-image-models
提交
91534522
P
pytorch-image-models
项目概览
stoneliu1981
/
pytorch-image-models
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
pytorch-image-models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
91534522
编写于
2月 01, 2020
作者:
R
Ross Wightman
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add newly added TF ported EfficientNet-B8 weights (RandAugment)
上级
82dd60b3
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
23 addition
and
9 deletion
+23
-9
timm/models/efficientnet.py
timm/models/efficientnet.py
+23
-9
未找到文件。
timm/models/efficientnet.py
浏览文件 @
91534522
...
...
@@ -124,6 +124,9 @@ default_cfgs = {
'tf_efficientnet_b7'
:
_cfg
(
url
=
'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b7_ra-6c08e654.pth'
,
input_size
=
(
3
,
600
,
600
),
pool_size
=
(
19
,
19
),
crop_pct
=
0.949
),
'tf_efficientnet_b8'
:
_cfg
(
url
=
'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b8_ra-572d5dd9.pth'
,
input_size
=
(
3
,
672
,
672
),
pool_size
=
(
21
,
21
),
crop_pct
=
0.954
),
'tf_efficientnet_b0_ap'
:
_cfg
(
url
=
'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_ap-f262efe1.pth'
,
mean
=
IMAGENET_INCEPTION_MEAN
,
std
=
IMAGENET_INCEPTION_STD
,
input_size
=
(
3
,
224
,
224
)),
...
...
@@ -1059,9 +1062,20 @@ def tf_efficientnet_b7(pretrained=False, **kwargs):
return
model
@
register_model
def
tf_efficientnet_b8
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B8. Tensorflow compatible variant """
# NOTE for train, drop_rate should be 0.5
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
'tf_efficientnet_b8'
,
channel_multiplier
=
2.2
,
depth_multiplier
=
3.6
,
pretrained
=
pretrained
,
**
kwargs
)
return
model
@
register_model
def
tf_efficientnet_b0_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B0. Tensorflow compatible variant """
""" EfficientNet-B0
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1071,7 +1085,7 @@ def tf_efficientnet_b0_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b1_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B1. Tensorflow compatible variant """
""" EfficientNet-B1
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1081,7 +1095,7 @@ def tf_efficientnet_b1_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b2_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B2. Tensorflow compatible variant """
""" EfficientNet-B2
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1091,7 +1105,7 @@ def tf_efficientnet_b2_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b3_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B3. Tensorflow compatible variant """
""" EfficientNet-B3
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1101,7 +1115,7 @@ def tf_efficientnet_b3_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b4_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B4. Tensorflow compatible variant """
""" EfficientNet-B4
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1111,7 +1125,7 @@ def tf_efficientnet_b4_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b5_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B5. Tensorflow compatible variant """
""" EfficientNet-B5
AdvProp
. Tensorflow compatible variant """
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
model
=
_gen_efficientnet
(
...
...
@@ -1121,7 +1135,7 @@ def tf_efficientnet_b5_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b6_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B6. Tensorflow compatible variant """
""" EfficientNet-B6
AdvProp
. Tensorflow compatible variant """
# NOTE for train, drop_rate should be 0.5
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
...
...
@@ -1132,7 +1146,7 @@ def tf_efficientnet_b6_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b7_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B7. Tensorflow compatible variant """
""" EfficientNet-B7
AdvProp
. Tensorflow compatible variant """
# NOTE for train, drop_rate should be 0.5
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
...
...
@@ -1143,7 +1157,7 @@ def tf_efficientnet_b7_ap(pretrained=False, **kwargs):
@
register_model
def
tf_efficientnet_b8_ap
(
pretrained
=
False
,
**
kwargs
):
""" EfficientNet-B
7
. Tensorflow compatible variant """
""" EfficientNet-B
8 AdvProp
. Tensorflow compatible variant """
# NOTE for train, drop_rate should be 0.5
kwargs
[
'bn_eps'
]
=
BN_EPS_TF_DEFAULT
kwargs
[
'pad_type'
]
=
'same'
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录